torch.FloatTensor()---于小勇

类型转换, 将list ,numpy转化为tensor。 以list -> tensor为例：

print(torch.FloatTensor([1,2])) # 输出: tensor([1., 2.])

根据torch.Size()创建一个空tensor

a = torch.tensor([[1, 2], [3, 4]]) print(torch.FloatTensor(a.size()))

输出如下

tensor([[ 9.4636e+33, 4.5559e-41], [-3.9725e-26, 3.0631e-41]]) ---来自腾讯云社区的---于小勇