pytorch Tensor 与 tensor

Tensor

torch.Tensor() 是类，不同的数据类型对应不同的类，一共有九种（图片来自于官方文档截图）：

torch.Tensor 是 torch.FloatTensor 的一个别名。不同类之间的转换可以用 type() 函数，如：

import torch

a = torch.Tensor([1, 2, 3])
b = a.type(torch.LongTensor)

tensor

torch.tensor() 是一个函数，它返回值是上面的某个类：

torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor

它通过dtype来确定是哪种数据类型。