torch.nanquantile
- torch.nanquantile(input, q, dim=None, keepdim=False, *, interpolation='linear', out=None) → Tensor
-
这是
torch.quantile()
的一个变体,它会“忽略”NaN
值,在计算分位数q
时,视作输入中的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])