mindspore.ops.nanmedian

mindspore.ops.nanmedian(input, axis=- 1, keepdims=False)

忽略NaN值，按指定轴计算输入tensor的中值和索引。

警告

如果 input 的中值不唯一，则 indices 不一定包含第一个出现的中值。

参数：

• input (Tensor) - 输入tensor。

• axis (int, 可选) - 指定轴，默认 -1 。

• keepdims (bool, 可选) - 输出tensor是否保持维度，默认 False 。

返回：

一个由tensor组成的tuple(median, median_indices)

支持平台：

CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[0.57, 0.11, float("nan")],
...             [0.38, float("nan"), float("nan")],
...             [0.36, 0.16, float("nan")]], mindspore.float32)
>>> y, idx = mindspore.ops.nanmedian(x, axis=0, keepdims=False)
>>> print(y)
[0.38 0.11  nan]
>>> print(idx)
[1 0 0]