Aminmax

Computes the minimum and maximum values of the input tensor.

Abstract Signature:

Aminmax(input: Tensor, dim: int, keepdim: bool = False)

PyTorch

API: torch.aminmax
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: —
Strategy: Macro '(jnp.min({input}, axis={dim}, keepdims={keepdim}), jnp.max({input}, axis={dim}, keepdims={keepdim}))'

NumPy

API: —
Strategy: Macro '(np.amin({input}, axis={dim}, keepdims={keepdim}), np.amax({input}, axis={dim}, keepdims={keepdim}))'

Keras

API: —
Strategy: Macro '(keras.ops.min({input}, axis={dim}, keepdims={keepdim}), keras.ops.max({input}, axis={dim}, keepdims={keepdim}))'

TensorFlow

API: —
Strategy: Macro '(tf.math.reduce_min({input}, axis={dim}, keepdims={keepdim}), tf.math.reduce_max({input}, axis={dim}, keepdims={keepdim}))'

Apple MLX

API: —
Strategy: Macro '(mx.min({input}, axis={dim}, keepdims={keepdim}), mx.max({input}, axis={dim}, keepdims={keepdim}))'

Flax NNX

API: —
Strategy: Macro '(jnp.min({input}, axis={dim}, keepdims={keepdim}), jnp.max({input}, axis={dim}, keepdims={keepdim}))'

PaxML / Praxis

API: —
Strategy: Macro '(jnp.min({input}, axis={dim}, keepdims={keepdim}), jnp.max({input}, axis={dim}, keepdims={keepdim}))'