理由如下:
torch.tensor() 和 torch.as_tensor() 的 API 更丰富,可控制的属性更多;
torch.Tensor() 会改变数据类型,torch.from_numpy(...out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) # 全为一个定值
参数说明:
*size:新张量的形状...”真的是是非常小的、接近 0 的数:
>>> torch.empty(1)
tensor([2.0890e+20])
还可以根据已有的张量,按照该张量的形状生成相同形状的新张量:
torch.zeros_like...轴聚合
tensor([[0, 1, 2, 0, 1, 2],
[3, 4, 5, 3, 4, 5]])
torch.stack(tensors, dim=0, out=None)
将两个张量叠加到一起...,每个比较返回一个布尔值,最终返回一个与被比较元素形状相同的张量:
torch.eq(input, other, out=None):如果 input 中的元素等于 output 中的对应元素,返回 True