tensoreflusso:: ops:: SoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Calcola il costo dell'entropia incrociata softmax e i gradienti per la propagazione all'indietro.
Riepilogo
Gli input sono i logit, non le probabilità.
Argomenti:
- scope: un oggetto Scope
- caratteristiche: matrice batch_size x num_classes
- etichette: matrice batch_size x num_classes Il chiamante deve garantire che ogni batch di etichette rappresenti una distribuzione di probabilità valida.
Resi:
- Perdita
Output
: perdita per esempio (vettore batch_size). - Backprop
Output
: gradienti retropropagati (matrice batch_size x num_classes).
Costruttori e distruttori | |
---|---|
SoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Attributi pubblici | |
---|---|
backprop | |
loss | |
operation |
Attributi pubblici
backprop
::tensorflow::Output backprop
perdita
::tensorflow::Output loss
operazione
Operation operation
Funzioni pubbliche
SoftmaxCrossEntropyWithLogits
SoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )