tensorflow::ops::SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

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

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

Аргументы:

  • scope: Объект Scope
  • характеристики: матрица batch_size x num_classes
  • метки: размер_пакета x количество_классов матрица Вызывающая сторона должна убедиться, что каждый пакет меток представляет собой допустимое распределение вероятностей.

Возвращает:

  • Output потери: потери для каждого примера (вектор batch_size).
  • Output данные обратного распространения: градиенты, полученные обратным распространением (матрица размером пакета x числом классов).

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

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
)