mlx.nn.log_softmax

class log_softmax(x, axis=-1)

应用对数 Softmax 函数。

逐元素应用 \(x + \log \sum_i e^{x_i}\)。