torch.FloatTensor()

2022-09-02 22:24:11 浏览数 (1)

类型转换, 将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]])

0 人点赞