torch.Tensor与torch.tensor的三个区别

torch.Tensor与torch.tensor的区别


区别一torch.Tensortorch.emptytorch.tensor的混合,在传入数据时,torch.Tensor使用全局默认类型(FloatTensor)而torch.tensor则会从输入数据中进行推断;

区别二:当输入数据是一个常数n时,torch.Tensor会将n是为一维张量的元素个数,并随机初始化。而torch.tensor则会将n视作一个数字而不是元素个数。例如:

torch.Tensor(6)

# tensor([2.7626e-06, 1.0871e-05, 3.4008e-06, 5.2903e-08, 2.1877e-04, 1.7377e-04])

torch.tensor(6)

# tensor(6)

区别三torch.Tensor是一个类,会调用自己的构造函数,而torch.tensor则是一个函数,其原型为:

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

猜你喜欢

转载自blog.csdn.net/qq_45957458/article/details/129656914