flux tensoriel : : opérations : : SoftmaxCrossEntropyWithLogits

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.

Retour:

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

Attributs publics

::tensorflow::Output backprop

::tensorflow::Output loss

Operation operation

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