tensorflow:: ops:: SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>Calcula el coste de entropía cruzada softmax y los gradientes para la retropropagación.
Resumen
A diferencia de SoftmaxCrossEntropyWithLogits , esta operación no acepta una matriz de probabilidades de etiquetas, sino una sola etiqueta por fila de características. Se considera que esta etiqueta tiene una probabilidad de 1.0 para la fila correspondiente.
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: vector batch_size con valores en [0, num_classes). Esta es la etiqueta para la entrada de minibatch dada.
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 | |
|---|---|
SparseSoftmaxCrossEntropyWithLogits (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
SparseSoftmaxCrossEntropyWithLogits
SparseSoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )