PyTorch简明教程torch.Tensor()与torch.tensor()的区别

Posted 2022-04-23 算法与编程之美

问题

当使用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的区别。