美文网首页
torch.Tensor 和 torch.tensor

torch.Tensor 和 torch.tensor

作者: 不太聪明的亚子 | 来源:发表于2020-05-13 10:09 被阅读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

相关文章

网友评论

      本文标题:torch.Tensor 和 torch.tensor

      本文链接:https://www.haomeiwen.com/subject/olmgnhtx.html