mindspore.ops.nanmean

mindspore.ops.nanmean(input, axis=None, keepdims=False, *, dtype=None)[源代码]

计算 input 指定维度元素的平均值，忽略NaN。如果指定维度中的所有元素都是NaN，则结果将是NaN。

参数：

  • input (Tensor) - 计算平均值的输入Tensor。

  • axis (int, 可选) - 求平均值的维度。默认值： None ，所有维度求平均值。

  • keepdims (bool, 可选) - 输出Tensor是否保持维度。默认值： False ，不保留维度。

关键字参数：

  • dtype (mindspore.dtype, 可选) - 输出Tensor的类型。默认值： None ，输出Tensor的类型和输入一致。

返回：

Tensor，输入 input 指定维度的元素平均值，忽略NaN。

异常：

  • TypeError - input 不是一个Tensor。

  • TypeError - axis 不是int类型。

  • TypeError - keepdims 不是bool类型。

  • TypeError - dtype 不是MindSpore的数据类型。

  • ValueError - axis 的范围不在[-r, r)，r 表示 input 的rank。

支持平台：

Ascend GPU CPU

样例：

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