Fonction torch.nanquantile de PyTorch

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

torch.nanquantileIl s'agit d'une fonction dans PyTorch utilisée pour renvoyer le quantile des valeurs non-NaN d'un tenseur. Contrairement à quantile, elle ignore toutes les valeurs NaN.

Définition de fonction

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

Exemple d'utilisation

Exemple

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])

# Renvoie le quantile 0.5 des valeurs non-NaN
print("Quantile 0.5 des valeurs non-NaN :", torch.nanquantile(x, 0.5))

# Renvoyer plusieurs quantiles non-NaN
print("Quantiles 0.25, 0.75 des valeurs non-NaN :", torch.nanquantile(x, torch.tensor([0.25, 0.75])))

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

Le résultat de sortie est:

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

Autres extensions