mlx.nn.log_softmax

class log_softmax(x, axis=-1)

套用對數 softmax 函式。

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