тензорный поток:: опс:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Вычисляет стоимость перекрестной энтропии softmax и градиенты для обратного распространения ошибки.

Краткое содержание

В отличие от SoftmaxCrossEntropyWithLogits , эта операция не принимает матрицу вероятностей меток, а принимает одну метку для каждой строки объектов. Считается, что эта метка имеет вероятность 1,0 для данной строки.

Входные данные — это логиты, а не вероятности.

Аргументы:

  • область: объект области .
  • функции: матрица Batch_size x num_classes
  • метки: вектор пакетного размера со значениями в [0, num_classes). Это метка для данной записи мини-пакета.

Возврат:

  • Output потери: потери в каждом примере (вектор размера пакета).
  • Output обратное распространение: градиенты с обратным распространением (матрица Batch_size x num_classes).

Конструкторы и деструкторы

SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels)

Публичные атрибуты

backprop
loss
operation

Публичные атрибуты

задняя опора

::tensorflow::Output backprop

потеря

::tensorflow::Output loss

операция

Operation operation

Общественные функции

SparseSoftmaxCrossEntropyWithLogits

 SparseSoftmaxCrossEntropyWithLogits(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input features,
  ::tensorflow::Input labels
)