tensorflow::ops::SparseSoftmaxCrossEntropyWithLogits
#include <nn_ops.h>Calcula o custo da entropia cruzada softmax e os gradientes para retropropagação.
Resumo
Diferentemente de SoftmaxCrossEntropyWithLogits , esta operação não aceita uma matriz de probabilidades de rótulos, mas sim um único rótulo por linha de características. Considera-se que este rótulo tem probabilidade 1,0 para a linha em questão.
Os parâmetros de entrada são os logits, não as probabilidades.
Argumentos:
- escopo: Um objeto de escopo
- 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 do minibatch em questão.
Devoluções:
- Perda
Output: Perda por exemplo (vetor batch_size). -
Outputretropropagação: gradientes retropropagados (matriz batch_size x num_classes).
Construtores e Destrutores | |
|---|---|
SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels) |
Atributos públicos | |
|---|---|
backprop | |
loss | |
operation | |
Atributos públicos
retropropagação
::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 )