pytorch学习笔记——torch.Tensor()与·torch.tensor()

torch.Tensor()是一个类，默认是torch.FloatTensor()的简称，创建的为float32位的数据类型；torch.tensor()是一个函数，是对张量数据的拷贝，根据传入data的类型来创建Tensor;a=torch.Tensor([1,2])type(a)#<class 'torch.Tensor'>a.type()#'torch.FloatTensor'a

phily123

2950人浏览 · 2021-09-11 13:51:47

phily123 · 2021-09-11 13:51:47 发布

torch.Tensor()是一个类，默认是torch.FloatTensor()的简称，创建的为float32位的数据类型；
torch.tensor()是一个函数，是对张量数据的拷贝，根据传入data的类型来创建Tensor;

a=torch.Tensor([1,2])
type(a)#<class 'torch.Tensor'>
a.type()#'torch.FloatTensor'

a = torch.tensor([1, 2])
type(a)#<class 'torch.Tensor'>
a.type()#'torch.LongTensor'

a=torch.Tensor([1,2])
a.dtype#torch.float32

a = torch.tensor([1, 2])
a.dtype#torch.int64
a = torch.tensor([1., 2.])
a.dtype#torch.float32