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	:: tensorflow::Output
loss	:: tensorflow::Output
operation	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
)