Pytorch view和permute的区别
a = torch.tensor([[[1,2,3],[4,5,6]]])
b = a.view(3,2)
c = a.permute(0,2,1)
print(a.size(),a)
print(b.size(),b)
print(c.size(),c)
输出结果不一样
a: torch.Size([1, 2, 3])
tensor([[[1, 2, 3],
[4, 5, 6]]])
b: torch.Size([3, 2])
tensor([[1, 2],
[3,
原创
2020-05-29 19:44:03 ·
862 阅读 ·
0 评论