PyTorch中的view函数可以改变tensor的形状
eg1
import torch
test = torch.zeros(2,3)
print(test)
print(test.view(3,-1)) #-1表示程序会根据上一个维度自动计算-1所在位置的维度
#tensor([[0., 0., 0.],
# [0., 0., 0.]])
#tensor([[0., 0.],
# [0., 0.],
# [0., 0.]])
eg2
test = torch.Tensor([1,2,3,4,5,6,7,8])
print(test)
print(test.view(4,2))
# tensor([1., 2., 3., 4., 5., 6., 7., 8.])
#tensor([[1., 2.],
# [3., 4.],
# [5., 6.],
# [7., 8.]])
网友评论