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