MaskedFill

Fills elements of self tensor with value where mask is True.

Abstract Signature:

MaskedFill(input: Tensor, mask: Tensor, value: float)

PyTorch

API: torch.masked_fill
Strategy: Infix (?)
Official Docs ↗

JAX (Core)

API: jnp.where
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.where
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.where
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.where
Strategy: Direct Mapping
Official Docs ↗