【PyTorch简明教程】torch.Tensor()与torch.tensor

【PyTorch简明教程】torch.Tensor()与torch.tensor,第1张

问题

当使用torch创建tensor的时候,发现以下两种方式均可:

a = torch.tensor([1,2,3])
b = torch.Tensor([4,5,6])

虽然提供了两种方式创建tensor,但是二者的区别是什么?

方法

torch.tensor()是一普通函数,而torch.Tensor()是类torch.Tensor的构造函数,此外在创建空tensor的时候,有一定的区别。

tensor_without_data = torch.Tensor()

# 当创建空的tensor的时候会报错
tensor_without_data = torch.tensor()
# 只能采用这种方式创建空tensor
tensor_without_data = torch.tensor(())
结语

本文主要介绍了torch.Tensor()和torch.tensor()两种方式创建tensor的区别。

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/714773.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-24
下一篇 2022-04-24

发表评论

登录后才能评论

评论列表(0条)

保存