mlx.nn.log_softmax

class log_softmax(x, axis=-1)

套用對數 softmax 函式。

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