PyTorch
API: torch.is_nonzero
Strategy: Direct Mapping
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'