tensorflow::ops::SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

Краткое описание

В отличие от SoftmaxCrossEntropyWithLogits, эта операция принимает не матрицу вероятностей меток, а одну метку для каждой строки признаков. Считается, что вероятность этой метки для соответствующей строки равна 1.0.

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

Аргументы:

• scope: объект Scope
• features: матрица размером batch_size x num_classes
• labels: вектор размера batch_size со значениями в диапазоне [0, num_classes). Метка для соответствующего элемента мини-пакета.

Возвращает:

• Output loss: функция потерь для каждого примера (вектор batch_size).
• Output backprop: градиенты обратного распространения ошибки (матрица размером batch_size x num_classes).

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

::tensorflow::Output backprop

::tensorflow::Output loss

Operation operation

Публичные функции

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