IsFiniteΒΆ

Returns a new tensor with boolean elements representing if each element is finite.

Abstract Signature:

IsFinite(input: Tensor)

PyTorch

API: torch.isfinite
Strategy: Direct Mapping

JAX (Core)

API: jnp.isfinite
Strategy: Direct Mapping

NumPy

API: np.isfinite
Strategy: Direct Mapping

Keras

API: keras.ops.isfinite
Strategy: Direct Mapping

TensorFlow

API: tf.math.is_finite
Strategy: Direct Mapping

Apple MLX

API: mlx.core.isfinite
Strategy: Macro '(~mx.isinf({input}) & ~mx.isnan({input}))'

Flax NNX

API: jnp.isfinite
Strategy: Direct Mapping

PaxML / Praxis

API: jnp.isfinite
Strategy: Direct Mapping