IsIn

Tests if each element of elements is in test_elements.

Abstract Signature:

IsIn(elements: Tensor, test_elements: Tensor, assume_unique: bool = False, invert: bool = False)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.isin

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jnp.isin

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: np.isin

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.ops.numpy.isin

Strategy: Macro 'keras.ops.convert_to_tensor(np.isin({elements}, {test_elements}, assume_unique={assume_unique}, invert={invert}))'

TensorFlow

API: tf.experimental.numpy.isin

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: jnp.isin

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jnp.isin

Strategy: Direct Mapping

Official Docs ↗