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.

Ritorna:

  • Perdita Output : perdita per esempio (vettore batch_size).
  • Backprop Output : gradienti propagati all'indietro (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
)