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