FullSoftmax¶

Applies Softmax function to an n-dimensional input Tensor, optionally computing CrossEntropy.

Abstract Signature:

FullSoftmax(dim: int = -1)

API: torch.nn.Softmax

Strategy: Direct Mapping

API: keras.layers.Softmax

Strategy: Direct Mapping

API: tf.keras.layers.Softmax

Strategy: Direct Mapping

API: mlx.nn.Softmax

Strategy: Direct Mapping

API: flax.nnx.softmax

Strategy: Direct Mapping

API: paxml.layers.FullSoftmax

Strategy: Direct Mapping