tf.nn.softmax

Вычисляет активации softmax.

Просмотреть псевдонимы

Основные псевдонимы

tf.math.softmax

tf.nn.softmax(
    logits, axis=None, name=None
)

Используется в блокнотах

Используется в руководствах	Используется в обучающих материалах
Многослойные перцептроны для распознавания цифр с помощью Core API Основы TensorFlow Введение в тензоры Обучение и оценка с помощью встроенных методов	Глубокое обучение с учетом неопределенности с помощью SNGP Интегрированные градиенты Пользовательское обучение: пошаговое руководство Классификация изображений Простое распознавание аудио: распознавание ключевых слов

Используется в руководствах

Используется в обучающих материалах

Используется для многоклассовых предсказаний. Сумма всех выходных значений, полученных с помощью softmax, равна 1.

Эта функция выполняет эквивалентную операцию

softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis, keepdims=True)

Пример использования:

softmax = tf.nn.softmax([-1, 0., 1.])
softmax
<tf.Tensor: shape=(3,), dtype=float32,
numpy=array([0.09003057, 0.24472848, 0.66524094], dtype=float32)>
sum(softmax)
<tf.Tensor: shape=(), dtype=float32, numpy=1.0>

Аргументы
`logits`	Непустой `Tensor`. Должен иметь один из следующих типов: `half`, `float32`, `float64`.
`axis`	Измерение, по которому будет выполняться softmax. По умолчанию равно -1, что указывает на последнее измерение.
`name`	Имя операции (необязательно).

Возвращает
`Tensor`. Имеет тот же тип и форму, что и `logits`.

Вызывает исключение
`InvalidArgumentError`	если `logits` пуст или `axis` выходит за пределы последнего измерения `logits`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/nn/softmax