view,将Tensor转成特定维数空间。
x = torch.zeros(2,2,3)
x.view(3,4) #等价于x:view(3, -1)
#-1 表示将剩余元素全部看成这一维度
输出:
tensor([[ 0., 0., 0., 0.],
[ 0., 0., 0., 0.],
[ 0., 0., 0., 0.]])
view,将Tensor转成特定维数空间。
x = torch.zeros(2,2,3)
x.view(3,4) #等价于x:view(3, -1)
#-1 表示将剩余元素全部看成这一维度
输出:
tensor([[ 0., 0., 0., 0.],
[ 0., 0., 0., 0.],
[ 0., 0., 0., 0.]])