Функция PyTorch torch.nanquantile

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

torch.nanquantileЭто функция в PyTorch, используемая для возврата квантиля значений тензора, не являющихся NaN. В отличие от quantile, она игнорирует все значения NaN.

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

torch.nanquantile(input, q, dim, keepdim=False)

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

Пример

import torch

x = torch.tensor([1.0, float('nan'), 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0])

# Вернуть квантиль 0.5 не-NaN значений
print("Квантиль 0.5 не-NaN значений:", torch.nanquantile(x, 0.5))

# Вернуть несколько не-NaN квантилей
print("Квантили 0.25, 0.75 не-NaN значений:", torch.nanquantile(x, torch.tensor([0.25, 0.75])))

# Не-NaN квантиль по dim=0
y = torch.tensor([[1.0, float('nan'), 3.0], [4.0, 5.0, 6.0]])
print("Квантиль 0.5 не-NaN значений при dim=0:", torch.nanquantile(y, 0.5, dim=0))

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

非NaN 0.5 分位数: tensor(5.5000)
非NaN 0.25, 0.75 分位数: tensor([2.7500, 7.7500])
dim=0 非NaN 0.5 分位数: tensor([2.5000, 5.0000, 4.5000])

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

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