fluxo tensor:: ops:: SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Calcula o custo de entropia cruzada softmax e gradientes para retropropagar.

Resumo

As entradas são logits, não probabilidades.

Argumentos:

  • escopo: um objeto Escopo
  • recursos: matriz batch_size x num_classes
  • rótulos: batch_size x num_classes matriz O chamador deve garantir que cada lote de rótulos represente uma distribuição de probabilidade válida.

Retorna:

  • Perda Output : perda por exemplo (vetor batch_size).
  • Backprop Output : gradientes retropropagados (matriz batch_size x num_classes).

Construtores e Destruidores

SoftmaxCrossEntropyWithLogits (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

SoftmaxCrossEntropyWithLogits

 SoftmaxCrossEntropyWithLogits(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input features,
  ::tensorflow::Input labels
)