NanStd

Compute the standard deviation along the specified axis, while ignoring NaNs.

Abstract Signature:

NanStd(a: Array, axis: int = None, dtype: Dtype = None, ddof: int = 0, keepdims: bool = False)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Macro 'torch.sqrt(torch.nanvar({a}, dim={axis}, correction={ddof}, keepdim={keepdims}))'

JAX (Core)

API: jax.numpy.nanstd

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.nanstd

Strategy: Direct Mapping

Official Docs ↗

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: tf.experimental.numpy.nanstd

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: jax.numpy.nanstd

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.nanstd

Strategy: Direct Mapping

Official Docs ↗