关于torch.FloatTensor()

类型转换, 将list ,numpy转化为tensor

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

posted @ 2021-03-15 16:28  吴莫愁258  阅读(2446)  评论(0)  编辑  收藏  举报