Fonction torch.mean de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.meanC'est une fonction dans PyTorch utilisée pour calculer la moyenne d'un tenseur.

Définition de fonction

torch.mean(input, dim, keepdim, dtype, out)

Exemples d'utilisation

Exemple

import torch

x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Calculer la moyenne de tous les éléments
print("Moyenne globale :", torch.mean(x))

# Calculer la moyenne selon dim=0
print("moyenne pour dim=0 :", torch.mean(x, dim=0))

Le résultat de sortie est:

全局均值: tensor(3.5000)
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions