torch.tensor与torch.Tensor区别

发布时间 2023-09-11 16:55:04作者: ChLizzZ

在pytorch中torch.tensor与torch.Tensor都可以生成新的张量

torch.Tensor是python类,会调用自己的构造函数,是默认张量类型torch.FloatTensor的别名,所以数据传入时会使用全局默认类型(FloatTensor)

而torch.tensor会根据输入的数据进行推断,根据原始数据类型生成相应的torch.LongTensor   torch.FloatTensor   torch.DoubleTensor  ,torch.tensor是一个函数

特别的:当输入数据是一个常数n时,torch.Tensor会将n视为一维张量的元素个数,并随机初始化

torch.tensor会将n视为一个数字   而不是元素个数