【Pytorch】torch.Tensor和torch.tensor的区别是什么？

160 阅读 0 评论 106 点赞

我是靠谱客的博主贤惠外套，这篇文章主要介绍【Pytorch】torch.Tensor和torch.tensor的区别是什么？，现在分享给大家，希望可以做个参考。

小写的tensor只接受现有的数据；而大写的Tensor相当于就是FloatTensor，既可以接收现有的数据，也可以接受shape来创建指定形状的Tensor。为了避免混淆，建议接收现有数据的时候使用tensor,指定shape的时候使用Tensor。

举例：

a = torch.tensor([2, 3.3]) #接受现有的数据
b = torch.Tensor([2, 3.3]) #等价于a = torch.FloatTensor([2, 3.3])
b = torch.FloatTensor(2, 3) #指定维度创建Tensor
c = torch.IntTensor(2, 3) #指定维度创建Tensor

a = np.array([2, 3.3])
b = torch.from_numpy(a)  # 注意：该种创建方式a与b共享内存，修改其中一个另外一个也会改变，需要注意。