一.dim的定義

dim的不同值代表不同的維度，例如在二維張量中dim=0代表的是行，dim=1代表的是列。廣泛的說，在多維張量()中，dim=0就是指，dim=n是指

二.例子

torch.sum()

?input：輸入的張量

dim:需要消減的維度

keepdim:輸出張量中是否保存指定dim維的張量

eg1:

b = torch.arange(3 * 2 * 2).view(3, 2, 2) print(b) print(torch.sum(b, (1, 2)))

輸出結果為：

這里的輸出結果是按照第0維進行相加的，原因是因為dim=(1, 2)將這兩維進行消減，從而根據剩下的一維進行求和計算。

eg2:

b = torch.arange(3 * 2 * 2).view(3, 2, 2) print(b) print(torch.sum(b, 1))

?

?這里將第一維消減后還有兩維，所以最終的輸出結果將按照第0維以及第2維進行計算，所以最終張量的維度為3*2.

總結

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

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