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 )