Fonction torch.nanmean de PyTorch

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

torch.nanmeanIl s'agit d'une fonction de PyTorch utilisée pour calculer la valeur moyenne en ignorant les valeurs NaN.

Définition de fonction

torch.nanmean(input, dim, keepdim=False, out=None)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur contenant NaN
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])

# Calculer la moyenne des valeurs non-NaN
mean = torch.nanmean(x)
print("Moyenne non-NaN :", mean)

Le résultat de sortie est:

非 NaN 均值: tensor(3.)

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

Autres extensions