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
: por ejemplo, pérdida (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 )