在使用pytorch中,有与numpy类似的转换数组形状的函数,view。
a = torch.arange(0,20)
a.view(4,5)
tensor([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14],
[15, 16, 17, 18, 19]])
某些维度也可以不指定数目
a.view(-1,5)
tensor([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14],
[15, 16, 17, 18, 19]])