Pytorch-Tensor和tensor的区别

Pytorch-Tensor和tensor的区别

torch.Tensortorch.FloatTensor的别名,是python类,只能创建FloatTensor。从首字母大写也可以看出一些端倪。

torch.tensor是一个函数,可以将其它数据结构转化成tensor,可以创建FloatTensorIntTensor等等。

例子:

>>> torch.Tensor([1, 2]).dtype
>>> torch.float32
>>> torch.tensor([1, 2]).dtype
>>> torch.int64
>>> torch.tensor([1, 2], dtype=torch.float32).dtype
>>> torch.float32
发布了173 篇原创文章 · 获赞 28 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/ECNU_LZJ/article/details/104203655