Функция PyTorch torch.nanmean

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.nanmeanЭто функция в PyTorch, используемая для вычисления среднего значения с игнорированием значений NaN.

Определение функции

torch.nanmean(input, dim, keepdim=False, out=None)

Пример использования

Пример

import torch

# Создать тензор, содержащий NaN
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])

# Вычислить среднее значение не-NaN значений
mean = torch.nanmean(x)
print("Среднее без NaN:", mean)

Результат вывода:

非 NaN 均值: tensor(3.)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения