问题

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