NanMeanΒΆ

Return the mean of array elements along a given axis, ignoring NaNs.

Abstract Signature:

NanMean(a: Tensor, axis: int | Sequence[int] | None, dtype: DType | None, keepdims: bool = False, where: Tensor | None)

PyTorch

API: torch.nanmean
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.nanmean
Strategy: Direct Mapping

NumPy

API: numpy.nanmean
Strategy: Direct Mapping