тензорный поток:: опс:: 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
)