Fonction torch.mode de PyTorch

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

torch.modeC'est une fonction de PyTorch utilisée pour renvoyer le mode d'un tenseur. Le mode est la valeur qui apparaît le plus fréquemment dans un ensemble de données.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer le mode
print("Mode :", torch.mode(x))

# Mode selon dim=0
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("mode selon dim=0 :", torch.mode(y, dim=0))

# Mode selon dim=1
print("mode selon dim=1 :", torch.mode(y, dim=1))

Le résultat de sortie est:

众数: torch.return_types.mode(values=tensor(3), indices=tensor(4))
dim=0 众数: torch.return_types.mode(values=tensor([1, 2, 2]), indices=tensor([0, 0, 0]))
dim=1 众数: torch.return_types.mode(values=tensor([2, 3]), indices=tensor([1, 2]))

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

Autres extensions