เทนเซอร์โฟลว์:: ออปส์:: อินท็อปเค

#include <nn_ops.h>

ระบุว่าเป้าหมายเหล่านั้นอยู่ในกลุ่มการคาดการณ์ K อันดับแรกหรือไม่

สรุป

ฟังก์ชันนี้จะส่งออกอาร์เรย์บูลีน batch_size โดยค่าใน out[i] จะเป็น true หากการทำนายสำหรับคลาสเป้าหมายอยู่ในกลุ่มการทำนาย k อันดับแรกจากทั้งหมดสำหรับตัวอย่าง i โปรดทราบว่าพฤติกรรมของ InTopK แตกต่างจาก TopK op ในการจัดการกับค่าที่เท่ากัน หากหลายคลาสมีค่าการทำนายเดียวกันและอยู่คร่อมขอบเขต k แรก คลาสเหล่านั้นทั้งหมดจะถือว่าอยู่ใน k อันดับแรก

กล่าวอย่างเป็นทางการยิ่งขึ้น ให้

\(predictions_i\) ให้การทำนายสำหรับทุกคลาส เช่น i , \(targets_i\) ตัวอย่างเช่น คลาสเป้าหมายคือ i , \(out_i\) ตัวอย่างเช่น i จะเป็นผลลัพธ์

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

ข้อโต้แย้ง:

  • ขอบเขต: วัตถุ ขอบเขต
  • การคาดการณ์: เทนเซอร์ batch_size x classes
  • เป้าหมาย: เวกเตอร์ batch_size ที่ประกอบด้วยรหัสคลาส
  • k: จำนวนองค์ประกอบสูงสุดที่ใช้ในการคำนวณความแม่นยำ

การส่งคืนสินค้า:

  • Output : ค่าความแม่นยำที่คำนวณได้ที่ k ในรูปแบบ bool Tensor

ตัวสร้างและตัวทำลาย

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

คุณลักษณะสาธารณะ

operation
precision

หน้าที่สาธารณะ

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

คุณลักษณะสาธารณะ

การดำเนินการ 

Operation operation

ความแม่นยำ 

::tensorflow::Output precision

หน้าที่สาธารณะ

อินท็อปเค 

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

โหนด 

::tensorflow::Node * node() const

ตัวดำเนินการ::เทนเซอร์โฟลว์::อินพุต 

 operator::tensorflow::Input() const

ตัวดำเนินการ::เทนเซอร์โฟลว์::เอาต์พุต 

 operator::tensorflow::Output() const