unsqueeze()函數起升維的作用,參數表示在哪個地方加一個維度。
在第一個維度(中括號)的每個元素加中括號
0表示在張量最外層加一個中括號變成第一維。
直接看例子:

import torch input=torch.arange(0,6) print(input) print(input.shape) 結果： tensor([0, 1, 2, 3, 4, 5]) torch.Size([6]) print(input.unsqueeze(0)) print(input.unsqueeze(0).shape) 結果： tensor([[0, 1, 2, 3, 4, 5]]) torch.Size([1, 6]) print(input.unsqueeze(1)) print(input.unsqueeze(1).shape) 結果： tensor([[0],[1],[2],[3],[4],[5]]) torch.Size([6, 1])

總結

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

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