torch.Tensor 和 torch.tensor
2020-05-13 本文已影响0人
不太聪明的亚子
创建一个张量,有时用torch.Tensor,有时用torch.tensor,大小写字母的区别到底在哪里。
torch.Tensor()是python类,生成的张量是默认数据类型:torch.FloatTensor(),即单精度浮点数类型;
torch.tensor()是python函数,生成的张量的数据类型,是根据括号中的数据的实际类型而定的,函数原型是:
torch.tensor(data, dtype=None, device=None, requires_grad=False)
如果torch.tensor([1,2,3]),那么数据类型是torch.LongTensor;
如果torch.tensor([1.0, 2.0, 3.0]),那么数据类型是torch.FloatTensor;
以此类推。
参考:
https://blog.csdn.net/tfcy694/article/details/85338745