PyTorch torch.nanquantile-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.nanquantileEs ist eine Funktion in PyTorch, die verwendet wird, um das Quantil der Nicht-NaN-Werte eines Tensors zurückzugeben. Im Gegensatz zu quantile ignoriert sie alle NaN-Werte.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

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-Quantil der Nicht-NaN-Werte zurückgeben
print("Nicht-NaN-0.5-Quantil:", torch.nanquantile(x, 0.5))

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

# Nicht-NaN-Quantil entlang dim=0
y = torch.tensor([[1.0, float('nan'), 3.0], [4.0, 5.0, 6.0]])
print("dim=0 Nicht-NaN-0.5-Quantil:", torch.nanquantile(y, 0.5, dim=0))

Das Ausgabeergebnis ist:

非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 Referenzhandbuch

Andere Erweiterungen