tf.nn.sparse_softmax_cross_entropy_with_logits
| Просмотреть исходный код на GitHub |
Вычисляет кросс-энтропию разреженного софтмакса между logits и labels.
tf.nn.sparse_softmax_cross_entropy_with_logits(
labels, logits, name=None
)
Измеряет ошибку вероятности в задачах дискретного классифицирования, в которых классы взаимоисключают (каждый элемент принадлежит ровно одному классу). Например, каждый образ CIFAR-10 имеет метку только одного класса: образ может быть собакой или грузовиком, но не тем и другим одновременно.
Примечание: Для этой операции вероятность заданного метки считается исключительной. То есть, мягкие классы не допускаются, и векторlabelsдолжен предоставлять единственный конкретный индекс для истинного класса для каждой строкиlogits(каждый элемент мини-пакета). Для классификации с мягким софтмаком и распределением вероятностей для каждого элемента см.softmax_cross_entropy_with_logits_v2.
Общее использование заключается в том, что логарифмы имеют форму [batch_size, num_classes] и метки имеют форму [batch_size], но поддерживаются более высокие измерения, в которых dim-е измерение предположительно имеет размер num_classes. logits должен иметь тип float16, float32, или float64, а labels должен иметь тип int32 или int64.
Обратите внимание, что для избежания путаницы необходимо передавать только именованные аргументы в эту функцию.
| Аргументы | |
|---|---|
labels | Tensor формы [d_0, d_1, ..., d_{r-1}] (где r — ранг labels и результата) и типа int32 или int64. Каждый элемент labels должен быть индексом в [0, num_classes). Другие значения приведут к исключению при выполнении этого оператора на процессоре и вернут NaN для соответствующих строк потерь и градиента на графическом процессоре. |
logits | Не масштабированные логарифмы вероятностей формы [d_0, d_1, ..., d_{r-1}, num_classes] и типа float16, float32, или float64. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Объект Tensor той же формы, что и labels и того же типа, что и logits с потерей кросс-энтропии софтмакса. |
| Исключения | |
|---|---|
ValueError | Если логарифмы являются скалярами (нужен ранг >= 1) или если ранг меток не равен рангу логарифмов минус один. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/nn/sparse_softmax_cross_entropy_with_logits