tensorflow:: ops:: SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Calcula el coste de entropía cruzada softmax y los gradientes para la retropropagación.

Resumen

Las entradas son los logits, no las probabilidades.

Argumentos:

  • ámbito: Un objeto de ámbito
  • Características: matriz de tamaño_lote x número_de_clases
  • etiquetas: matriz tamaño_lote x número_clases El llamador 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 batch_size).
  • Retropropagación Output : gradientes retropropagados (matriz de tamaño de lote x número de clases).

Constructores y destructores

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

Atributos públicos

backprop
loss
operation

Atributos públicos

Retropropulsión

::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
)