MaskedSelect

Returns a new 1-D tensor which indexes the input tensor according to the boolean mask.

Abstract Signature:

MaskedSelect(input: Tensor, mask: Tensor)

PyTorchJAX (Core)NumPyTensorFlow

PyTorch

API: torch.masked_select

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{input}[{mask}]'

NumPy

API: —

Strategy: Macro '{input}[{mask}]'

TensorFlow

API: tf.boolean_mask

Strategy: Direct Mapping

Official Docs ↗