mlx.nn.softmax

class softmax(x, axis=-1)

应用 Softmax 函数。

逐元素应用 \(\frac{e^{x_i}}{\sum_j e^{x_j}}\)。