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

上一篇 下一篇

猜你喜欢

热点阅读