tf.keras.activations.softmax

Softmax преобразует вектор значений в распределение вероятностей.

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

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

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

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

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

Вводные значения представляют собой логарифмы вероятностей.