Función torch.median de PyTorch

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

torch.medianEs una función en PyTorch que se utiliza para devolver la mediana de un tensor. La mediana es el valor que se encuentra en el medio después de ordenar los datos.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve la mediana de todos los elementos
print("Mediana global:", torch.median(x))

# Mediana a lo largo de dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Mediana en dim=0:", torch.median(y, dim=0))

El resultado de salida es:

全局中位数: tensor(3)
dim=0 中位数: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))

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

Otras extensiones