torch.tensor和torch.Tensor的区别

时间:2024-03-13 10:31:43

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

一、torch.tensor( )

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

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

二、torch.Tensor( )

torch.Tensor是一个class,是torch.FloatTensor的别称
torch.tensor和torch.Tensor的区别
在pycharm上显示如下:
torch.tensor和torch.Tensor的区别
示例如下:将list转化为一个tensor
torch.tensor和torch.Tensor的区别

三、对比:

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