Функция torch.quantile в PyTorch

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

torch.quantileявляется функцией в PyTorch, используемой для возврата квантиля тензора. Квантиль — это значение, которое делит данные пропорционально.

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

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

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

Пример

import torch

x = torch.tensor([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

# Вернуть квантиль 0.5 (медиану)
print("0.5 quantile:", torch.quantile(x, 0.5))

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

# Квантиль вдоль dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("dim=0 0.5 квантиль:", torch.quantile(y, 0.5, dim=0))

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

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

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

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