Fonction torch.quantile de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.quantileest une fonction de PyTorch utilisée pour renvoyer le quantile d'un tenseur. Un quantile est une valeur qui divise les données proportionnellement.

Définition de fonction

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

Exemples d'utilisation

Exemple

import torch

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

# Renvoie le quantile 0.5 (médiane)
print("0.5 quantile:", torch.quantile(x, 0.5))

# Renvoie plusieurs quantiles
print("quantiles 0.25, 0.75 :", torch.quantile(x, torch.tensor([0.25, 0.75])))

# Quantile selon dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("quantile 0.5 selon dim=0 :", torch.quantile(y, 0.5, dim=0))

Le résultat de sortie est:

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 参考手册Manuel de référence Pytorch torch

Autres extensions