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

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

สรุป

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

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

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

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

ข้อโต้แย้ง:

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

คุณลักษณะเสริม (ดูที่ 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
)