mindspore.ops.nanmean

mindspore.ops.nanmean(input, axis=None, keepdims=False, *, dtype=None)

忽略NaN值，按指定轴计算输入tensor的平均值。

参数：

• input (Tensor) - 输入tensor。

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

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

关键字参数：

• dtype (mindspore.dtype, 可选) - 返回的数据类型，默认 None 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[0.5, -1.1, float('nan')], [3.4, float('nan'), float('nan')]], mindspore.float32)
>>> y = mindspore.ops.nanmean(x, axis=0, keepdims=False)
>>> print(y)
[ 1.95 -1.1    nan]