fluxo tensor:: ops:: SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Calcula o custo de entropia cruzada softmax e gradientes para retropropagar.
Resumo
Ao contrário de SoftmaxCrossEntropyWithLogits
, esta operação não aceita uma matriz de probabilidades de rótulo, mas sim um único rótulo por linha de recursos. Este rótulo é considerado como tendo probabilidade 1,0 para a linha fornecida.
As entradas são logits, não probabilidades.
Argumentos:
- escopo: um objeto Escopo
- recursos: matriz batch_size x num_classes
- rótulos: vetor batch_size com valores em [0, num_classes). Este é o rótulo para a entrada do minilote fornecida.
Retorna:
- Perda
Output
: perda por exemplo (vetor batch_size). - Backprop
Output
: gradientes retropropagados (matriz batch_size x num_classes).
Construtores e Destruidores | |
---|---|
SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Atributos públicos | |
---|---|
backprop | |
loss | |
operation |
Atributos públicos
apoio traseiro
::tensorflow::Output backprop
perda
::tensorflow::Output loss
operação
Operation operation
Funções públicas
SparseSoftmaxCrossEntropyWithLogits
SparseSoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )