torch.nanquantile¶
- torch.nanquantile(input, q, dim=None, keepdim=False, *, interpolation='linear', out=None) Tensor ¶
这是
torch.quantile()
的一个变体,它“忽略”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])