tf.keras.activations.softmax

Softmax преобразует вещественный вектор в вектор категориальных вероятностей.

tf.keras.activations.softmax(
    x, axis=-1
)

Элементы выходного вектора находятся в диапазоне (0, 1) и суммируются до 1.

Каждый вектор обрабатывается независимо. Аргумент axis задаёт, вдоль какой оси входных данных применяется функция.

Softmax часто используется в качестве активации для последнего слоя классификационной сети, потому что результат можно интерпретировать как распределение вероятностей.

Softmax каждого вектора x вычисляется как exp(x) / tf.reduce_sum(exp(x)).

Входные значения являются логарифмическими вероятностями результата.