torch.nanquantile

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

这是torch.quantile()的一个变体,它会“忽略”NaN值,在计算分位数q时,视作输入中的NaN值不存在。如果某一行的所有值都是NaN,则该行的分位数也将是NaN。请参阅torch.quantile()的相关文档。

参数
  • input (Tensor) – 需要输入的张量。

  • q (floatTensor) – 包含范围在 [0, 1] 内的分位数值的标量或1D张量

  • dim (int) - 需要减少的维度。

  • keepdim (bool) – 是否在输出张量中保留dim维度。

关键字参数
  • interpolation (str) – 当所需的分位数位于两个数据点之间时,指定插值方法。可选的插值方法包括 linear, lower, higher, midpointnearest。默认为 linear

  • out (Tensor, 可选) – 指定输出张量。

示例:

>>> 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])
本页目录