Fonction torch.median de PyTorch
Manuel de référence PyTorch torch
torch.medianC'est une fonction dans PyTorch utilisée pour renvoyer la médiane d'un tenseur. La médiane est la valeur centrale après tri des données.
Définition de fonction
torch.median(input, dim, keepdim=False)
Exemple d'utilisation
Exemple
import torch
x = torch.tensor([4, 2, 1, 3, 5])
# Retourner la médiane de tous les éléments
print("Médiane globale :", torch.median(x))
# Médiane selon dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("médiane dim=0 :", torch.median(y, dim=0))
x = torch.tensor([4, 2, 1, 3, 5])
# Retourner la médiane de tous les éléments
print("Médiane globale :", torch.median(x))
# Médiane selon dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("médiane dim=0 :", torch.median(y, dim=0))
Le résultat de sortie est:
全局中位数: tensor(3) dim=0 中位数: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))
Autres extensions