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
)