mindspore.ops.nanmean
- mindspore.ops.nanmean(input, axis=None, keepdims=False, *, dtype=None)[source]
Computes the mean of input in specified dimension, ignoring NaN. If all elements in the specified dimensions are NaN, the result will be NaN.
- Parameters
- Keyword Arguments
dtype (mindspore.dtype, optional) – The output Tensor data type. Default:
None
, the data type of output Tensor is same as the input.- Returns
Tensor, the mean of input input in the given dimension axis, while ignoring NaNs.
- Raises
TypeError – If input is not a Tensor.
TypeError – If axis is not int.
TypeError – If keepdims is not bool.
TypeError – If dtype is not mindspore dtype.
ValueError – If axis is not in range of [-r, r) which r means the rank of input.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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]