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

#include <nn_ops.h>

ค้นหาค่าและดัชนีขององค์ประกอบที่ใหญ่ที่สุด 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

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

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)

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

indices
operation
values

ฟังก์ชันคงที่สาธารณะ

Sorted (bool x)

โครงสร้าง

เทนเซอร์โฟลว์ :: ops :: TopK :: Attrs

ตัวตั้งค่าคุณลักษณะเพิ่มเติมสำหรับ TopK

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

ดัชนี

::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
)