aliran tensor:: operasi:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Menghitung biaya entropi silang softmax dan gradien ke propagasi mundur.

Ringkasan

Berbeda dengan SoftmaxCrossEntropyWithLogits , operasi ini tidak menerima matriks probabilitas label, melainkan satu label per baris fitur. Label ini dianggap memiliki probabilitas 1,0 untuk baris tertentu.

Inputnya adalah logit, bukan probabilitas.

Argumen:

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

Pengembalian:

  • Kerugian Output : Per contoh kerugian (vektor ukuran_batch).
  • Backprop Output : gradien yang dipropagasi mundur (matriks batch_size x num_classes).

Konstruktor dan Destruktor

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
)