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 )