舉一個非常簡單直觀的例子：

首先，利用torch創建一個矩陣A：

????????A=torch.tensor([[1,2,3],[2,3,4]])

矩陣A的大小為：2*3

（1）dim=0:

??????? B=torch.mean(A.float(),dim=0)

此時打印B為：

?????? tensor([1.5000, 2.5000, 3.5000])???? #說明dim=0時，是豎著計算均值，即按列計算；

(2) dim=1:

??????? C=torch.mean(A.float(),dim=1)

此時打印C為：

??????? tensor([2., 3.])??????????????????????????????? #說明dim=1時，是橫向計算均值，即按行計算.

????????

????????

????????????????

總結

以上是生活随笔為你收集整理的PYTORCH中torch.mean(),dim=0,dim=1分别代表什么的全部內容，希望文章能夠幫你解決所遇到的問題。

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