SoftminΒΆ

Applies the Softmin function.

Abstract Signature:

Softmin(x: Tensor, axis: int = -1)

PyTorch

API: torch.nn.functional.softmin
Strategy: Direct Mapping

JAX (Core)

API: jax.nn.softmin
Strategy: Direct Mapping

Keras

API: β€”
Strategy: Macro 'keras.activations.softmax(-{x}, axis={axis})'

TensorFlow

API: tf.nn.softmin
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.softmin
Strategy: Direct Mapping

Flax NNX

API: jax.nn.softmin
Strategy: Direct Mapping

PaxML / Praxis

API: jax.nn.softmin
Strategy: Direct Mapping