tensorflow::ops::SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>Вычисляет стоимость кросс-энтропии softmax и градиенты для обратного распространения ошибки.
Краткое содержание
В отличие от SoftmaxCrossEntropyWithLogits , эта операция принимает не матрицу вероятностей меток, а одну метку на каждую строку признаков. Считается, что вероятность этой метки для данной строки равна 1,0.
Inputs are the logits, not probabilities.
Аргументы:
- scope: Объект Scope
- features: batch_size x num_classes matrix
- labels: batch_size — вектор со значениями в диапазоне [0, num_classes). Это метка для данной записи мини-пакета.
Возвращает:
-
Outputпотери: потери для каждого примера (вектор batch_size). -
Outputданные обратного распространения: градиенты, полученные обратным распространением (матрица размером пакета x числом классов).
Конструкторы и Деструкторы | |
|---|---|
SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Общедоступные атрибуты | |
|---|---|
backprop | |
loss | |
operation | |
Общедоступные атрибуты
обратное распространение
::tensorflow::Output backprop
потеря
::tensorflow::Output loss
операция
Operation operation
Общественные функции
SparseSoftmaxCrossEntropyWithLogits
SparseSoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )