tensorflow::ops::SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

Сводка

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

Аргументы:

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

Возвращает:

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

Открытые атрибуты

::tensorflow::Output backprop

::tensorflow::Output loss

Operation operation

Открытые функции

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