mlx.nn.log_softmax

class log_softmax(x, axis=-1)

Applies the Log Softmax function.

Applies \(x + \log \sum_i e^{x_i}\) element wise.