1）flatten(x,1)是按照x的第1個維度拼接（按照列來拼接，橫向拼接）；
2）flatten(x,0)是按照x的第0個維度拼接（按照行來拼接，縱向拼接）；
3）有時候會遇到flatten里面有兩個維度參數，flatten(x, start_dim, end_dimension)，此時flatten函數執行的功能是將從start_dim到end_dim之間的所有維度值乘起來，其他的維度保持不變。例如x是一個size為[4,5,6]的tensor, flatten(x, 0, 1)的結果是一個size為[20,6]的tensor。

In [1]: import torchIn [2]: A = torch.tensor([[[1,2,3,4],[5,6,7,8],[9,10,11,12]],[[13,14,15,16],[17,18,19,20],[21,22,23,24]]]) In [3]: A.size Out[3]: <function Tensor.size>In [4]: A.shape Out[4]: torch.Size([2, 3, 4])In [5]: B = torch.flatten(A,1)In [6]: B Out[6]: tensor([[ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],[13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24]])In [7]: B.shape Out[7]: torch.Size([2, 12])In [8]: C = torch.flatten(A,0,1)In [9]: C Out[9]: tensor([[ 1, 2, 3, 4],[ 5, 6, 7, 8],[ 9, 10, 11, 12],[13, 14, 15, 16],[17, 18, 19, 20],[21, 22, 23, 24]])In [10]: C.shape Out[10]: torch.Size([6, 4])

總結

以上是生活随笔為你收集整理的torch.flatten()函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。