Fonction torch.nanmedian de PyTorch
Manuel de référence PyTorch torch
torch.nanmedianIl s'agit d'une fonction de PyTorch utilisée pour renvoyer la médiane des valeurs non-NaN dans un tenseur. Contrairement à median, elle ignore toutes les valeurs NaN.
Définition de fonction
torch.nanmedian(input, dim, keepdim=False)
Exemple d'utilisation
Exemple
import torch
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# Retourner la médiane des valeurs non-NaN
print("Médiane non-NaN :", torch.nanmedian(x))
# Médiane non-NaN selon dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("Médiane non-NaN pour dim=0 :", torch.nanmedian(y, dim=0))
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# Retourner la médiane des valeurs non-NaN
print("Médiane non-NaN :", torch.nanmedian(x))
# Médiane non-NaN selon dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("Médiane non-NaN pour dim=0 :", torch.nanmedian(y, dim=0))
Le résultat de sortie est:
非NaN中位数: tensor(2.) dim=0 非NaN中位数: torch.return_types.nanmedian(values=tensor([1., 1., 2.]), indices=tensor([0, 1, 0]))
Autres extensions