tensorflow :: operaciones :: SoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Calcula el costo de la entropía cruzada de Softmax y los gradientes para propagar hacia atrás.
Resumen
Las entradas son los logits, no las probabilidades.
Argumentos:
- alcance: un objeto de alcance
- características: tamaño de lote x matriz num_clases
- etiquetas: tamaño_lote x matriz num_clases El llamador debe asegurarse de que cada lote de etiquetas representa una distribución de probabilidad válida.
Devoluciones:
- Pérdida de
Output
: Por ejemplo, pérdida (vector batch_size). - Backprop de
Output
: gradientes retropropagados (batch_size x num_classes matrix).
Constructores y Destructores | |
---|---|
SoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Atributos públicos | |
---|---|
backprop | |
loss | |
operation |
Atributos públicos
backprop
::tensorflow::Output backprop
pérdida
::tensorflow::Output loss
operación
Operation operation
Funciones publicas
SoftmaxCrossEntropyWithLogits
SoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )