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 )