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

#include <nn_ops.h>

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

สรุป

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

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

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