เทนเซอร์โฟลว์:: ปฏิบัติการ:: อินท็อปเควี2

#include <nn_ops.h>

บอกว่าเป้าหมายอยู่ในการทำนาย K อันดับต้น ๆ หรือไม่

สรุป

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

อย่างเป็นทางการมากขึ้น, ให้

predictionsi เป็นการทำนายสำหรับทุกคลาสเช่น i targetsi เป็นคลาสเป้าหมายเช่น i outi เป็นผลลัพธ์เช่น i

outi=predictionsi,targetsiTopKIncludingTies(predictionsi)

ข้อโต้แย้ง:

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

ผลตอบแทน:

  • Output : ความแม่นยำที่คำนวณได้ที่ k เป็น bool Tensor

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

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

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

operation
precision

งานสาธารณะ

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

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

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

Operation operation

ความแม่นยำ

::tensorflow::Output precision

งานสาธารณะ

อินท็อปเควี2

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

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ::tensorflow::อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ::tensorflow::เอาต์พุต

 operator::tensorflow::Output() const