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