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

#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	:: tensorflow::Output
operation	Operation
values	:: tensorflow::Output

ฟังก์ชันคงที่สาธารณะ
Sorted (bool x)	Attrs

โครงสร้าง
เทนเซอร์โฟลว์ :: 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
)