tensorflow::ops::SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

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

В отличие от 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
)