PyTorch torch.quantile-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.quantileist eine Funktion in PyTorch, die verwendet wird, um das Quantil eines Tensors zurückzugeben. Ein Quantil ist ein Wert, der Daten proportional teilt.

Funktionsdefinition

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

Anwendungsbeispiele

Beispiel

import torch

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

# Rückgabe des 0.5-Quantils (Median)
print("0.5-Quantil:", torch.quantile(x, 0.5))

# Mehrere Quantile zurückgeben
print("0.25, 0.75-Quantile:", torch.quantile(x, torch.tensor([0.25, 0.75])))

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

Das Ausgabeergebnis ist:

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 Referenzhandbuch

Andere Erweiterungen