tensorflow :: ops :: InTopK

#include <nn_ops.h>

Menyatakan apakah target ada dalam prediksi K teratas.

Ringkasan

Ini mengeluarkan array bool batch_size , entri out[i] bernilai true jika prediksi untuk kelas target termasuk di antara prediksi k teratas di antara semua prediksi misalnya i . Perhatikan bahwa perilaku InTopK berbeda dari op TopK dalam menangani hubungan; jika beberapa kelas memiliki nilai prediksi yang sama dan mengangkang top k batas, semua kelas-kelas dianggap di atas k .

Lebih formal, biarkan

\(predictions_i\) adalah prediksi untuk semua kelas misal i , \(targets_i\) jadilah kelas target misal i , \(out_i\) jadilah keluaran misal i ,

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

Argumen:

  • scope: Objek Scope
  • prediksi: batch_size x classes .
  • target: Vektor batch_size id kelas.
  • k: Jumlah elemen teratas yang harus diperhatikan untuk presisi komputasi.

Pengembalian:

Pembuat dan Penghancur

InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 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

InTopK

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

simpul

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Keluaran

 operator::tensorflow::Output() const