ส่งกลับค่า k ต่ำสุด/สูงสุด และดัชนีของตัวถูกดำเนินการอินพุตในลักษณะโดยประมาณ
ดูhttps://arxiv.org/abs/2206.14286สำหรับรายละเอียดอัลกอริทึม การดำเนินการนี้ได้รับการปรับให้เหมาะกับ TPU เท่านั้นในปัจจุบัน
คลาสที่ซ้อนกัน
ระดับ | ApproxTopK.ตัวเลือก | คุณลักษณะเพิ่มเติมสำหรับ ApproxTopK |
วิธีการสาธารณะ
ApproxTopK.Options แบบคงที่ | รวมToTopk (บูลีนรวมToTopk) |
คงที่ <T ขยายจำนวน> ApproxTopK <T> | สร้าง (ขอบเขต ขอบเขต , อินพุต Operand <T>, Long k, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ ApproxTopK ใหม่ |
เอาต์พุต <จำนวนเต็ม> | ดัชนี () ดัชนีของ "ค่า" ควบคู่ไปกับ "การลดขนาด" ของตัวถูกดำเนินการ "อินพุต" |
ApproxTopK.Options แบบคงที่ | isMaxK (บูลีน isMaxK) |
ApproxTopK.Options แบบคงที่ | RecallTarget (การเรียกคืนเป้าหมายแบบลอยตัว) |
ApproxTopK.Options แบบคงที่ | ลดขนาด (ลดขนาดยาว) |
ApproxTopK.Options แบบคงที่ | ลดInputSizeOverride (ลดยาวInputSizeOverride) |
เอาท์พุต <T> | ค่า () ค่า k ต่ำสุด/สูงสุดตาม `reduction_ dimensions` ของตัวถูกดำเนินการ `input` |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะ ApproxTopK.Options รวม ToTopk (บูลีนรวม ToTopk)
พารามิเตอร์
รวมToTopk | เมื่อเป็นจริง จะรวมผลลัพธ์โดยประมาณเข้ากับ top-k เมื่อเป็นเท็จ จะส่งกลับผลลัพธ์โดยประมาณ จำนวนผลลัพธ์โดยประมาณคือการดำเนินการที่กำหนดไว้ และมากกว่าเท่ากับ `k` ที่ระบุ |
---|
สร้าง ApproxTopK <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต Operand <T>, Long k, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ ApproxTopK ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ป้อนข้อมูล | อาร์เรย์เพื่อค้นหา ต้องมีอย่างน้อย 1-D ของประเภทลอยตัว |
เค | ระบุจำนวนของ min/max-k |
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ ApproxTopK
ดัชนี เอาท์พุต สาธารณะ <จำนวนเต็ม> ()
ดัชนีของ "ค่า" ควบคู่ไปกับ "การลดขนาด" ของตัวถูกดำเนินการ "อินพุต"
สาธารณะคงที่ ApproxTopK.Options isMaxK (บูลีน isMaxK)
พารามิเตอร์
isMaxK | เมื่อเป็นจริง ให้คำนวณ max-k; มิฉะนั้นจะคำนวณ min-k |
---|
สาธารณะ ApproxTopK.Options เรียกคืนเป้าหมาย (โฟลตเรียกคืนเป้าหมาย)
พารามิเตอร์
เรียกคืนเป้าหมาย | เรียกคืนเป้าหมายสำหรับการประมาณ ช่วงใน (0,1] |
---|
สาธารณะคงที่ ApproxTopK.Options ลดขนาด (ลดขนาดยาว)
พารามิเตอร์
การลดขนาด | มิติข้อมูลจำนวนเต็มที่จะค้นหา ค่าเริ่มต้น: -1 |
---|
สาธารณะคงที่ ApproxTopK.Options ลดInputSizeOverride (ลดยาวInputSizeOverride)
พารามิเตอร์
ลดInputSizeOverride | เมื่อตั้งค่าเป็นค่าบวก จะแทนที่ขนาดที่กำหนดโดย `อินพุต [reduction_dim]` เพื่อประเมินการเรียกคืน ตัวเลือกนี้มีประโยชน์เมื่อ "อินพุต" ที่กำหนดเป็นเพียงส่วนย่อยของการคำนวณโดยรวมใน SPMD หรือไปป์ไลน์แบบกระจาย โดยที่ขนาดอินพุตที่แท้จริงไม่สามารถเลื่อนออกไปได้ด้วยรูปร่าง "อินพุต" |
---|
สาธารณะ เอาท์พุท <T> ค่า ()
ค่า k ต่ำสุด/สูงสุดตาม `reduction_ dimensions` ของตัวถูกดำเนินการ `input` มิติข้อมูลจะเหมือนกับตัวถูกดำเนินการ `อินพุต` ยกเว้น `มิติการลด`: เมื่อ `aggregate_to_topk` เป็นจริง มิติการลดขนาดจะเป็น `k`; มิฉะนั้นจะมากกว่าเท่ากับ `k` โดยที่ขนาดถูกกำหนดไว้ในการใช้งาน