tf.compat.v1.nn.softmax_cross_entropy_with_logits_v2
Вычисляет перекрёстную энтропию softmax между logits и labels. (устаревшие аргументы)
tf.compat.v1.nn.softmax_cross_entropy_with_logits_v2(
labels, logits, axis=None, name=None, dim=None
)
Используется в блокнотах
| Используется в обучающих материалах |
|---|
Измеряет ошибку вероятности в задачах дискретной классификации, в которых классы взаимоисключающие (каждый элемент принадлежит ровно одному классу). Например, каждое изображение CIFAR-10 помечено одной и только одной меткой: изображение может быть собакой или грузовиком, но не тем и другим одновременно.
Примечание: Хотя классы взаимоисключающие, их вероятности не обязательно должны быть таковыми. Требуется лишь, чтобы каждая строка labels представляла собой корректное распределение вероятностей. В противном случае вычисление градиента будет некорректным.
Если используются взаимоисключающие labels (когда в каждый момент времени истинен только один класс), см. sparse_softmax_cross_entropy_with_logits.
Распространённый вариант использования — логиты и метки формы [batch_size, num_classes], однако поддерживаются и более высокие размерности; размерность класса задаётся аргументом axis.
logits и labels должны иметь одинаковый тип данных (float16, float32 или float64).
Обратное распространение будет выполняться как для logits, так и для labels. Чтобы запретить обратное распространение для labels, перед передачей в эту функцию пропустите тензоры меток через tf.stop_gradient.
Обратите внимание: во избежание путаницы в эту функцию необходимо передавать только именованные аргументы.
| Аргументы | |
|---|---|
labels | Каждый вектор вдоль размерности класса должен представлять собой корректное распределение вероятностей. Например, если метки имеют форму [batch_size, num_classes], каждая строка labels[i] должна представлять собой корректное распределение вероятностей. |
logits | Нескорректированные логарифмы вероятностей. |
axis | Размерность класса. По умолчанию равна -1, то есть используется последняя размерность. |
name | Имя операции (необязательно). |
dim | Устаревший псевдоним для axis. |
| Возвращает | |
|---|---|
Tensor, содержащий значение потерь перекрёстной энтропии softmax. Его тип совпадает с типом 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/api_docs/python/tf/compat/v1/nn/softmax_cross_entropy_with_logits_v2
