类型转换, 将list ,numpy转化为tensor。 以list -> tensor为例:
代码语言:javascript复制print(torch.FloatTensor([1,2]))
# 输出: tensor([1., 2.])
根据torch.Size()创建一个空tensor
代码语言:javascript复制a = torch.tensor([[1, 2], [3, 4]])
print(torch.FloatTensor(a.size()))
输出如下
代码语言:javascript复制tensor([[ 9.4636e 33, 4.5559e-41],
[-3.9725e-26, 3.0631e-41]])