IsInfΒΆ

Tests if each element of input is infinite.

Abstract Signature:

IsInf(input: Tensor)

PyTorch

API: torch.isinf
Strategy: Direct Mapping

JAX (Core)

API: jnp.isinf
Strategy: Direct Mapping

NumPy

API: np.isinf
Strategy: Direct Mapping

Keras

API: keras.ops.isinf
Strategy: Direct Mapping

TensorFlow

API: tf.math.is_inf
Strategy: Direct Mapping

Apple MLX

API: mx.isinf
Strategy: Direct Mapping

Flax NNX

API: jnp.isinf
Strategy: Direct Mapping

PaxML / Praxis

API: jnp.isinf
Strategy: Direct Mapping