Función torch.nanmean de PyTorch
Manual de referencia de PyTorch torch
torch.nanmeanEs una función de PyTorch utilizada para calcular el promedio ignorando los valores NaN.
Definición de la función
torch.nanmean(input, dim, keepdim=False, out=None)
Ejemplo de uso
Ejemplo
import torch
# Crear un tensor que contiene NaN
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# Calcular el promedio de los valores no NaN
mean = torch.nanmean(x)
print("Promedio no NaN:", mean)
# Crear un tensor que contiene NaN
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# Calcular el promedio de los valores no NaN
mean = torch.nanmean(x)
print("Promedio no NaN:", mean)
El resultado es:
非 NaN 均值: tensor(3.)
Otras extensiones