Función torch.nansum de PyTorch

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

torch.nansumEs una función de PyTorch que se utiliza para devolver la suma de los valores no NaN de un tensor. A diferencia de sum, ignora todos los valores NaN.

Definición de la función

torch.nansum(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 suma de los valores no NaN
print("Suma de valores no NaN:", torch.nansum(x))

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

El resultado de salida es:

非NaN值之和: tensor(11.)
dim=0 非NaN值之和: tensor([5., 1., 5.])
dim=1 非NaN值之和: tensor([3., 8.])

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

Otras extensiones