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)