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).
  • Output retropropagaçã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
)