aliran tensor:: operasi:: DiTopKV2

#include <nn_ops.h>

Mengatakan apakah target berada dalam prediksi K teratas.

Ringkasan

Ini menghasilkan array bool batch_size , entri out[i] true jika prediksi untuk kelas target termasuk di antara k prediksi teratas di antara semua prediksi misalnya i . Perhatikan bahwa perilaku InTopK berbeda dengan operasi TopK dalam menangani ikatan; jika beberapa kelas memiliki nilai prediksi yang sama dan mengangkangi batas k teratas, semua kelas tersebut dianggap berada di k teratas.

Lebih formalnya, biarlah

\(predictions_i\) menjadi prediksi untuk semua kelas misalnya i , \(targets_i\) menjadi kelas target misalnya i , \(out_i\) menjadi output misalnya i ,

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

Argumen:

  • ruang lingkup: Objek Lingkup
  • prediksi: Tensor classes batch_size x.
  • target: Vektor batch_size dari id kelas.
  • k : Jumlah elemen teratas yang harus diperhatikan untuk presisi komputasi.

Pengembalian:

  • Output : Presisi yang dihitung pada k sebagai bool Tensor .

Konstruktor dan Destruktor

InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k)

Atribut publik

operation
precision

Fungsi publik

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atribut publik

operasi

Operation operation

presisi

::tensorflow::Output precision

Fungsi publik

DiTopKV2

 InTopKV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  ::tensorflow::Input k
)

simpul

::tensorflow::Node * node() const

operator::tensorflow::Masukan

 operator::tensorflow::Input() const

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const