view(-1)的作用

一般來說view()就是重新改變張量的維度，但如果view()里只有一個 -1，是什么意思呢？

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

結果如下圖：

結論

X.view(-1)中的-1本意是根據(jù)另外一個數(shù)來自動調整維度，但是這里只有一個維度，因此就會將X里面的所有維度數(shù)據(jù)轉化成一維的，并且按先后順序排列。