тензорный поток:: опс:: SoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Вычисляет стоимость перекрестной энтропии softmax и градиенты для обратного распространения ошибки.
Краткое содержание
Входные данные — это логиты, а не вероятности.
Аргументы:
- область: объект области.
- функции: матрица Batch_size x num_classes
- метки: матрица размер_пакета x число_классов. Вызывающий объект должен гарантировать, что каждый пакет меток представляет собой допустимое распределение вероятностей.
Возврат:
-
Output
потери: потери в каждом примере (вектор размера пакета). -
Output
обратное распространение: градиенты с обратным распространением (матрица Batch_size x num_classes).
Конструкторы и деструкторы | |
---|---|
SoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Публичные атрибуты | |
---|---|
backprop | |
loss | |
operation |
Публичные атрибуты
задняя опора
::tensorflow::Output backprop
потеря
::tensorflow::Output loss
операция
Operation operation
Общественные функции
SoftmaxCrossEntropyWithLogits
SoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )