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