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