torch.tensor和torch.Tensor的区别

torch.tensor()和torch.Tensor()的区别

一、torch.tensor( )

torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor

torch.tensor( )是一个function，在pycharm使用的时候会有标志
在使用的时候根据传入的data进行建造一个tensor

二、torch.Tensor( )

torch.Tensor是一个class，是torch.FloatTensor的别称

在pycharm上显示如下：

示例如下：将list转化为一个tensor

三、对比：

通过上面两张图对比，区别在于得到的tensor
通过torch.tensor( )得到的tensor的数据是int型的，是根据data的类型构造
torch.Tensor得到的是float型