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

#include <nn_ops.h>

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

สรุป

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

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

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

หากสององค์ประกอบเท่ากันองค์ประกอบดัชนีล่างจะปรากฏขึ้นก่อน

อาร์กิวเมนต์:

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

แอตทริบิวต์เสริม (ดู Attrs ):

  • เรียงลำดับ: หากเป็นจริงองค์ประกอบ 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)

โครงสร้าง

tensorflow :: ops :: TopK :: Attrs

ตัวตั้งค่าแอตทริบิวต์เสริมสำหรับ TopK

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

ดัชนี

::tensorflow::Output indices

การดำเนินการ

Operation operation

ค่า

::tensorflow::Output values

หน้าที่สาธารณะ

TopK

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

TopK

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

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

จัดเรียง

Attrs Sorted(
  bool x
)