aliran tensor:: operasi:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Menghitung biaya softmax lintas entropi dan gradien untuk backpropagate.

Ringkasan

Tidak seperti SoftmaxCrossEntropyWithLogits , operasi ini tidak menerima matriks probabilitas label, melainkan satu label per baris fitur. Label ini dianggap memiliki probabilitas 1,0 untuk baris yang diberikan.

Input adalah logit, bukan probabilitas.

Argumen:

  • ruang lingkup: Objek Lingkup
  • fitur: matriks batch_size x num_classes
  • labels: vektor batch_size dengan nilai dalam [0, num_classes). Ini adalah label untuk entri minibatch yang diberikan.

Pengembalian:

  • Kehilangan Output : Per contoh kerugian (vektor batch_size).
  • Output backprop: gradien backpropagated (batch_size x num_classes matrix).

Konstruktor dan Destructor

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

Atribut publik

backprop
loss
operation

Atribut publik

penyangga punggung

::tensorflow::Output backprop

kehilangan

::tensorflow::Output loss

operasi

Operation operation

Fungsi publik

SparseSoftmaxCrossEntropyWithLogits

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