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)).

Вводные значения — это логарифмы вероятностей полученного результата.