Función torch.mode de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.modeEs la función en PyTorch que devuelve la moda de un tensor. La moda es el valor que aparece con mayor frecuencia en un conjunto de datos.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve la moda
print(Moda:, torch.mode(x))

# Moda a lo largo de dim=0
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print(Moda dim=0:, torch.mode(y, dim=0))

# Moda a lo largo de dim=1
print(Moda dim=1:, torch.mode(y, dim=1))

El resultado es:

众数: 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 参考手册Manual de referencia de Pytorch torch

Otras extensiones