tf.keras.ops.softmax

Функция активации softmax.

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

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

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

Определение:

f(x) = exp(x) / sum(exp(x))

Пример:

x = np.array([-1., 0., 1.])
x_softmax = keras.ops.softmax(x)
print(x_softmax)
array([0.09003057, 0.24472847, 0.66524096], shape=(3,), dtype=float64)