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