Función torch.nanmedian de PyTorch

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

torch.nanmedianEs una función en PyTorch que se utiliza para devolver la mediana de los valores no NaN de un tensor. A diferencia de median, ignora todos los valores NaN.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])

# Devuelve la mediana de los valores no NaN
print("Mediana no NaN:", torch.nanmedian(x))

# Mediana no NaN a lo largo de dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("Mediana no NaN de dim=0:", torch.nanmedian(y, dim=0))

El resultado de salida es:

非NaN中位数: tensor(2.)
dim=0 非NaN中位数: torch.return_types.nanmedian(values=tensor([1., 1., 2.]), indices=tensor([0, 1, 0]))

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

Otras extensiones