对pytorch中x = x.view(x.size(0), -1) 的理解说明

在pytorch的CNN代码中经常会看到首先，在pytorch中的view()函数就是用来改变tensor的形状的，例如将2行3列的tensor变为1行6列，其中-1表示会自适应的调整剩余的维度在CNN中卷积或者池化之后需要连接全连接层，所以需要把多维度的tensor展平成一维，x.view就实现的这个功能卷积或者池化之后的tensor的维度为，其中x.size指batchsize的值，最后通过x.view将tensor的结构转换为了，即将拉直，然后就可以和fc层连接了view在pytorch中是用来改变张量的shape的，简单又好用。pytorch中view的用法通常是直接在张量名后用.view调用，然后放入自己想要的shape。从这里我们可以看得出来，view函数如其名，只改变“看起来”的样子，不会改变张量在内存中的排列。