Aminmax ======= Computes the minimum and maximum values of the input tensor. **Abstract Signature:** ``Aminmax(input: Tensor, dim: int, keepdim: bool = False)`` .. raw:: html

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}))'