一、方法詳解

含義：顧名思義，返回一個Tensor的均值

torch.mean(input, dim, keepdim=False)

• input：輸入的張量

• dim：求均值的維度，若dim = 0，則沿行求均值；若dim=1，則沿列求均值

• keepdim：輸出張量是否跟輸入張量的另一個維度相同

這個用言語很難說明白，我們直接通過案例來理解掌握！

二、案例

• 案例1

import torch x = torch.randn(4, 4) print(x.size()) y = torch.mean(x, dim=0, keepdim=True) print(y.size())

• 分析

如果我們要沿dim=0這個維度相同，而且keepdim=True，那么輸出張量的另一個維度dim=1就跟輸入張量相同。

• 運行結果

• 案例2

import torch x = torch.randn(4, 4) print(x.size()) y = torch.mean(x, dim=1, keepdim=True) print(y.size())

• 運行結果

總結

以上是生活随笔為你收集整理的PyTorch基础（九）----- torch.mean()方法的全部內容，希望文章能夠幫你解決所遇到的問題。

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