tensorflow::ops::SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Вычисляет стоимость кросс-энтропии софтмакс и градиенты для обратного распространения.
Краткое описание
В отличие от SoftmaxCrossEntropyWithLogits, эта операция не принимает матрицу вероятностей меток, а принимает единственную метку на строку признаков. Эта метка считается имеющей вероятность 1.0 для данной строки.
На вход подаются логарифмы, а не вероятности.
Аргументы:
- scope: Объект Scope
- features: матрица batch_size x num_classes
- labels: вектор batch_size со значениями в [0, num_classes). Это метка для данной записи минибатча.
Возвращает:
-
Outputloss: Потеря на пример (вектор batch_size). -
Outputbackprop: градиенты обратного распространения (матрица batch_size x num_classes).
| Конструкторы и деструкторы | |
|---|---|
SparseSoftmaxCrossEntropyWithLogits(const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels) |
| Общедоступные атрибуты | |
|---|---|
backprop | |
loss | |
operation | |
Общедоступные атрибуты
backprop
::tensorflow::Output backprop
loss
::tensorflow::Output loss
operation
Operation operation
Общедоступные функции
SparseSoftmaxCrossEntropyWithLogits
SparseSoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input 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/cc/class/tensorflow/ops/sparse-softmax-cross-entropy-with-logits