tensorflow::ops::SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Calcula o custo da entropia cruzada softmax e os gradientes para retropropagação.

Resumo

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: batch_size x num_classes matrix O chamador deve garantir que cada lote de rótulos represente uma distribuição de probabilidade válida.

Devoluções:

  • Perda Output : Perda por exemplo (vetor batch_size).
  • Output retropropagação: gradientes retropropagados (matriz batch_size x num_classes).

Construtores e Destrutores

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

Entropia cruzada softmax com logitos 

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