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

ค้นหาค่าและดัชนีขององค์ประกอบที่ใหญ่ที่สุด k รายการสำหรับมิติสุดท้าย

สรุป

หากอินพุตเป็นเวกเตอร์ (อันดับ 1) ให้ค้นหารายการที่ใหญ่ที่สุด k รายการในเวกเตอร์ แล้วส่งออกค่าและดัชนีเป็นเวกเตอร์ ดังนั้น values[j] จึงเป็นรายการที่ใหญ่ที่สุดอันดับ j ใน input และดัชนีของมันคือ indices[j]

สำหรับเมทริกซ์ (การตอบสนองอินพุตอันดับสูงกว่า) คำนวณรายการ k สูงสุดในแต่ละแถว (เวกเตอร์การตอบสนองตามมิติสุดท้าย) ดังนั้น,

values.shape = indices.shape = input.shape[:-1] + [k]

หากองค์ประกอบสองรายการเท่ากัน องค์ประกอบดัชนีต่ำกว่าจะปรากฏก่อน

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• อินพุต: 1-D หรือสูงกว่าโดยมีมิติสุดท้ายอย่างน้อย k
• เค: 0-D จำนวนองค์ประกอบบนสุดที่จะค้นหาตามมิติสุดท้าย (ตามแต่ละแถวสำหรับเมทริกซ์)

แอ็ตทริบิวต์ทางเลือก (ดู Attrs ):

• sorted: หากเป็นจริง ผลลัพธ์ k องค์ประกอบจะถูกจัดเรียงตามค่าจากมากไปน้อย

ผลตอบแทน:

• ค่า Output : องค์ประกอบที่ใหญ่ที่สุด k รายการตามแต่ละส่วนมิติสุดท้าย
• ดัชนี Output : ดัชนีของ values ภายในมิติสุดท้ายของ input

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

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)