Pytorch可以通过torch.tensor和torch.Tensor生成张量,现将其部分区别记录如下:
1、torch.tensor
Python的列表或序列可以通过torch.tensor()函数构造张量。张量的维度可以通过.shape查看,并可使用.size()方法计算张量的形状大小,使用.numel()方法计算张量中包含元素的数量。
在使用torch.tensor()函数时,可以使用参数dtype来指定张量的数据类型,使用参数requires_grad来指定张量是否需要计算梯度。
2、torch.Tensor
也可以使用torch.Tensor来生成张量,且可以根据指定的形状来生成张量。但是不能指定张量的数据类型。