NanSum ====== Returns the sum of all elements, treating NaNs as zero. **Abstract Signature:** ``NanSum(input: Tensor, dim: int, keepdim: bool = False)`` .. raw:: html

PyTorch

API: torch.nansum
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.nansum
Strategy: Direct Mapping

NumPy

API: numpy.nansum
Strategy: Direct Mapping

Keras

API:
Strategy: Custom / Partial

TensorFlow

API: tf.experimental.numpy.nansum
Strategy: Direct Mapping

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API: jax.numpy.nansum
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.nansum
Strategy: Direct Mapping