PutMask ======= Changes elements of an array based on conditional and input values. **Abstract Signature:** ``PutMask(a: Array, mask: Array, values: Array)`` .. raw:: html

PyTorch

API:
Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.putmask
Strategy: Direct Mapping

NumPy

API: numpy.putmask
Strategy: Direct Mapping

Keras

API:
Strategy: Custom / Partial

TensorFlow

API:
Strategy: Custom / Partial

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API: jax.numpy.putmask
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.putmask
Strategy: Direct Mapping