flux tensoriel : : opérations : : SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Calcule le coût d'entropie croisée softmax et les gradients à rétropropager.

Résumé

Les entrées sont les logits, pas les probabilités.

Arguments :

• scope : un objet Scope
• fonctionnalités : matrice batch_size x num_classes
• labels : matrice batch_size x num_classes L'appelant doit s'assurer que chaque lot d'étiquettes représente une distribution de probabilité valide.

Retours :

• Perte Output : perte par exemple (vecteur batch_size).
• Backprop Output : dégradés rétropropagés (matrice batch_size x num_classes).

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

Attributs publics
backprop	:: tensorflow::Output
loss	:: tensorflow::Output
operation	Operation

Attributs publics

arrière-plan

::tensorflow::Output backprop

perte

::tensorflow::Output loss

opération

Operation operation

Fonctions publiques

SoftmaxCrossEntropyWithLogits

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