Python PyTorch nanquantile用法及代碼示例

本文簡要介紹python語言中 torch.nanquantile 的用法。

用法:

torch.nanquantile(input, q, dim=None, keepdim=False, *, out=None) → Tensor

參數：

• input(Tensor) -輸入張量。

• q(float或者Tensor) -[0, 1] 範圍內的分位數值的標量或一維張量

• dim(int) -要減少的維度。

• keepdim(bool) -輸出張量是否保留了dim。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

這是 torch.quantile() 的變體，“ignores” NaN 值，計算分位數q，就好像input 中的NaN 值不存在一樣。如果縮減行中的所有值都是 NaN ，那麽該縮減的分位數將是 NaN 。請參閱 torch.quantile() 的文檔。

例子：

>>> t = torch.tensor([float('nan'), 1, 2])
>>> t.quantile(0.5)
tensor(nan)
>>> t.nanquantile(0.5)
tensor(1.5000)
>>> t = torch.tensor([[float('nan'), float('nan')], [1, 2]])
>>> t
tensor([[nan, nan],
        [1., 2.]])
>>> t.nanquantile(0.5, dim=0)
tensor([1., 2.])
>>> t.nanquantile(0.5, dim=1)
tensor([   nan, 1.5000])