tf.nn.softmax
Вычисляет активации softmax.
tf.nn.softmax(
logits, axis=None, name=None
)
Эта функция выполняет эквивалент
softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis)
| Аргументы |
|---|
logits | Непустой Tensor. Должен быть одного из следующих типов: half, float32, float64. |
axis | Размерность, по которой будет выполняться softmax. По умолчанию -1, что указывает на последнюю размерность. |
name | Имя операции (необязательно). |
| Возвращаемое значение |
|---|
A Tensor. Имеет тот же тип и форму, что и logits. |
| Исключения |
|---|
InvalidArgumentError | если logits пуста или axis выходит за пределы последней размерности logits. |
