PyTorch torch.nanquantile 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.nanquantileは、PyTorch でテンソルの非NaN値の分位数を返すために使用される関数です。quantile と異なり、すべての NaN 値を無視します。

関数定義

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

使用例

実例

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

# 非NaN値の0.5分位数を返す
print("非NaN 0.5 分位数:", torch.nanquantile(x, 0.5))

# 複数の非NaN分位数を返す
print("非NaN 0.25, 0.75 分位数:", torch.nanquantile(x, torch.tensor([0.25, 0.75])))

# dim=0 に沿った非NaN分位数
y = torch.tensor([[1.0, float('nan'), 3.0], [4.0, 5.0, 6.0]])
print("dim=0 非NaN 0.5 分位数:", torch.nanquantile(y, 0.5, dim=0))

出力結果は次のとおりです：

非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 リファレンスマニュアル

その他の拡張