flujo tensor:: operaciones:: SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Calcula el costo de entropía cruzada de softmax y los gradientes para propagar hacia atrás.

Resumen

Las entradas son logits, no probabilidades.

Argumentos:

  • alcance: un objeto de alcance
  • características: matriz de tamaño de lote x núm_clases
  • etiquetas: matriz de tamaño de lote x núm_clases La persona que llama debe asegurarse de que cada lote de etiquetas represente una distribución de probabilidad válida.

Devoluciones:

  • Pérdida Output : pérdida por ejemplo (vector de tamaño de lote).
  • Backprop Output : gradientes retropropagados (matriz de tamaño de lote x num_classes).

Constructores y destructores

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

Atributos públicos

backprop
loss
operation

Atributos públicos

respaldo

::tensorflow::Output backprop

pérdida

::tensorflow::Output loss

operación

Operation operation

Funciones públicas

SoftmaxCrossEntropyWithLogits

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