IsNonZero

Returns True if the input is a single element tensor which is not equal to zero.

Abstract Signature:

IsNonZero(input: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.is_nonzero

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{input}.size == 1 and {input}.item() != 0'

NumPy

API: numpy.all

Strategy: Macro '{input}.size == 1 and {input}.item() != 0'

Keras

API: —

Strategy: Macro 'keras.ops.size({input}) == 1 and keras.ops.convert_to_numpy({input}).item() != 0'

TensorFlow

API: —

Strategy: Macro 'tf.size({input}) == 1 and tf.get_static_value({input}) != 0'

Apple MLX

API: —

Strategy: Macro '{input}.size == 1 and {input}.item() != 0'

Flax NNX

API: —

Strategy: Macro '{input}.size == 1 and {input}.item() != 0'

PaxML / Praxis

API: —

Strategy: Macro '{input}.size == 1 and {input}.item() != 0'