tensorflow::ops::SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

Résumé

Les données d'entrée sont les logits, et non les probabilités.

Arguments :

  • portée : un objet Scope
  • caractéristiques : matrice taille_lot x nombre_classes
  • étiquettes : 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).
  • Rétropropagation Output : gradients 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
loss
operation

attributs publics

contre-poussée

::tensorflow::Output backprop

perte

::tensorflow::Output loss

opération

Operation operation

fonctions publiques

SoftmaxCrossEntropieAvecLogits

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