flujo tensor:: operaciones:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

Resumen

A diferencia de SoftmaxCrossEntropyWithLogits , esta operación no acepta una matriz de probabilidades de etiquetas, sino una única etiqueta por fila de características. Se considera que esta etiqueta tiene una probabilidad de 1,0 para la fila dada.

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: vector de tamaño de lote con valores en [0, num_clases). Esta es la etiqueta de la entrada del minibatch determinada.

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

SparseSoftmaxCrossEntropyWithLogits (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

SparseSoftmaxCrossEntropyWithLogits

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