tf.compat.v1.nn.softmax_cross_entropy_with_logits
Вычисляет перекрестную энтропию софтмакса между logits и labels. (устаревшее)
tf.compat.v1.nn.softmax_cross_entropy_with_logits(
_sentinel=None, labels=None, logits=None, dim=-1, name=None, axis=None
)
В будущих основных версиях TensorFlow градиенты будут передаваться в входные метки в процессе обратного распространения по умолчанию.
См. tf.nn.softmax_cross_entropy_with_logits_v2.
Измеряет ошибку вероятности в задачах дискретного классифицирования, в которых классы взаимоисключающие (каждая запись принадлежит ровно одному классу). Например, каждое изображение CIFAR-10 помечено одним и только одним ярлыком: изображение может быть собакой или грузовиком, но не тем и другим.
Примечание: Хотя классы взаимно исключают друг друга, их вероятности необязательно таковы. Все, что требуется, это то, что каждая строка labels представляет собой действительное распределение вероятностей. Если это не так, вычисление градиента будет неверным.
Если используется взаимоисключающее labels (где один и только один класс является истинным в каждый момент времени), см. sparse_softmax_cross_entropy_with_logits.
Частым случаем использования является наличие логарифмов и меток формы [batch_size, num_classes], но поддерживаются и более высокие размерности, при этом аргумент dim указывает размерность класса.
Обратное распространение произойдёт только в logits. Чтобы вычислить потерю перекрестной энтропии, которая позволяет обратное распространение в logits и labels, см. tf.nn.softmax_cross_entropy_with_logits_v2.
Обратите внимание, что для избежания путаницы требуется передавать только именованные аргументы в эту функцию.
| Args | |
|---|---|
_sentinel | Используется для предотвращения позиционных параметров. Внутреннее, не использовать. |
labels | Каждый вектор вдоль размерности класса должен содержать допустимое распределение вероятностей, например, в случае, когда метки имеют форму [batch_size, num_classes], каждая строка labels[i] должна быть допустимым распределением вероятностей. |
logits | Активации на метку, обычно линейный выход. Эти энергии активации интерпретируются как ненормализованные логарифмические вероятности. |
dim | Размерность класса. По умолчанию -1, что соответствует последней размерности. |
name | Имя операции (необязательно). |
axis | Псевдоним для dim. |
| Returns | |
|---|---|
A Tensor, который содержит потерю перекрестной энтропии софтмакса. Его тип такой же, как у logits, а форма такая же, как у labels, за исключением того, что у него нет последней размерности labels. |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/nn/softmax_cross_entropy_with_logits
