torch库view函数用法

1函数.view()

view（）用于改变tensor的形状

>>>x = torch.ones(2,2)
>>>xv = x.view(-1)
>>>x
tensor([[1.,1.],
        [1.,1.]])
>>>xv
tensor([1.,1.,1.，1.])

将一个2×2的张量变为1×4，就是通过view(1,4)，1行4列。如果改为view（4，1），就会变为4×1的tensor,参数-1表示自适应，比如长度4的向量，view(-1,1),就相当于view（4，1）。