torch.nanmean

torch.nanmean(input, dim=None, keepdim=False, *, dtype=None, out=None) → Tensor

Вычисляет среднее значение всех non-NaN элементов по указанным измерениям.

Эта функция идентична torch.mean(), когда в тензоре input нет NaN значений. При наличии NaN, torch.mean() будет распространять NaN на выход, а torch.nanmean() будет игнорировать NaN значения (torch.nanmean(a) эквивалентно torch.mean(a[~a.isnan()])).

Если keepdim True, тензор вывода имеет тот же размер, что и input, за исключением измерения(измерений) dim, где он имеет размер 1. В противном случае dim сжимается (см. torch.squeeze()), в результате чего тензор вывода имеет на 1 (или len(dim)) меньше измерение(измерений).

Параметры

• input (Tensor) – входной тензор.
• dim (int или tuple целых чисел, необязательно) – измерение или измерения для сокращения. Если None, все измерения сокращаются.
• keepdim (bool) – сохраняется ли dim размер измерения в выходном тензоре.

Ключевые аргументы

• dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. Если указано, входной тензор приводится к dtype перед выполнением операции. Это полезно для предотвращения переполнения типов данных. По умолчанию: None.
• out (Tensor, необязательно) – тензор вывода.

См. также

torch.mean() вычисляет среднее значение, распространяя NaN.

Пример:

>>> x = torch.tensor([[torch.nan, 1, 2], [1, 2, 3]])
>>> x.mean()
tensor(nan)
>>> x.nanmean()
tensor(1.8000)
>>> x.mean(dim=0)
tensor([   nan, 1.5000, 2.5000])
>>> x.nanmean(dim=0)
tensor([1.0000, 1.5000, 2.5000])

# If all elements in the reduced dimensions are NaN then the result is NaN
>>> torch.tensor([torch.nan]).nanmean()
tensor(nan)