Función torch.quantile de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.quantileEs una función de PyTorch que se utiliza para devolver los cuantiles de un tensor. Los cuantiles son valores que dividen los datos en proporciones.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve el cuantil 0.5 (mediana)
print("Cuantil 0.5:", torch.quantile(x, 0.5))

# Devuelve múltiples cuantiles
print("Cuantiles 0.25, 0.75:", torch.quantile(x, torch.tensor([0.25, 0.75])))

# Cuantil a lo largo de dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("Cuantil 0.5 para dim=0:", torch.quantile(y, 0.5, dim=0))

El resultado de salida es:

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 参考手册Manual de referencia de Pytorch torch

Otras extensiones