tensorflow:: ops:: SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>
Calcula o custo de entropia cruzada softmax e gradientes para retropropagação.
Resumo
Ao contrário de SoftmaxCrossEntropyWithLogits
, esta operação não aceita uma matriz de probabilidades de rótulos, 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 os logits, não as probabilidades.
Argumentos:
- scope: um objeto Scope
- características: matriz batch_size x num_classes
- labels: vetor batch_size com valores em [0, num_classes). Este é o rótulo para a entrada de minilote fornecida.
Devoluções:
- Perda de
Output
: Por exemplo, perda (vetor batch_size). -
Output
backprop: 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
backprop
::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 )