view(-1)的作用

一般来说view()就是重新改变张量的维度，但如果view()里只有一个 -1，是什么意思呢？

import torch
a = torch.randn(3,5,2)
print(a)
print(a.view(-1))

结果如下图：

结论

X.view(-1)中的-1本意是根据另外一个数来自动调整维度，但是这里只有一个维度，因此就会将X里面的所有维度数据转化成一维的，并且按先后顺序排列。