tf.compat.v1.losses.sparse_softmax_cross_entropy

Функция потерь кросс-энтропии, использующая tf.nn.sparse_softmax_cross_entropy_with_logits.

tf.compat.v1.losses.sparse_softmax_cross_entropy(
    labels, logits, weights=1.0, scope=None, loss_collection=tf.GraphKeys.LOSSES,
    reduction=Reduction.SUM_BY_NONZERO_WEIGHTS
)

weights действует как коэффициент для функции потерь. Если указано скалярное значение, то функция потерь просто масштабируется заданным значением. Если weights является тензором формы [batch_size], то веса потерь применяются к каждой соответствующей выборке.

Аргументы
`labels`	`Tensor` формы `[d_0, d_1, ..., d_{r-1}]` (где `r` — ранг `labels` и результата) и типа `int32` или `int64`. Каждый элемент в `labels` должен быть индексом в `[0, num_classes)`. Другие значения приведут к исключению при выполнении этого оператора на процессоре CPU и возвращению `NaN` для соответствующих строк потерь и градиентов на GPU.
`logits`	Немасштабированные логарифмические вероятности формы `[d_0, d_1, ..., d_{r-1}, num_classes]` и типа `float16`, `float32` или `float64`.
`weights`	Коэффициенты для функции потерь. Это должно быть скалярное значение или значение, которое может быть расширено до `labels` (то есть того же ранга и каждой размерностью, которая либо равна 1, либо одинаковая).
`scope`	область видимости операций, выполняемых при вычислении функции потерь.
`loss_collection`	коллекция, в которую будет добавлена функция потерь.
`reduction`	Тип сокращения, применяемого к функции потерь.

Возвращаемое значение
Функция потерь с весами `Tensor` того же типа, что и `logits`. Если `reduction` равно `NONE`, это имеет ту же форму, что и `labels`; в противном случае это скаляр.

Исключения
`ValueError`	Если формы `logits`, `labels`, и `weights` несовместимы или если какие-либо из них равны None.

Совместимость с Eager Execution

Аргумент loss_collection игнорируется при выполнении в режиме Eager. Подумайте о том, чтобы сохранить возвращаемое значение или собрать потери с помощью tf.keras.Model.

© 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.3/api_docs/python/tf/compat/v1/losses/sparse_softmax_cross_entropy