מחזירה ערכי min/max k והמדדים שלהם של אופרנד הקלט באופן משוער.
ראה https://arxiv.org/abs/2206.14286 לפרטי האלגוריתם. אופציה זו מותאמת רק ב-TPU כרגע.
כיתות מקוננות
מעמד | ApproxTopK.Options | תכונות אופציונליות עבור ApproxTopK |
שיטות ציבוריות
סטטי ApproxTopK.Options | aggregateToTopk (בוליאני aggregateToTopk) |
סטטי <T מרחיב מספר> ApproxTopK <T> | ליצור ( היקף היקף , קלט Operand <T>, Long k, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת ApproxTopK חדשה. |
פלט <מספר שלם> | מדדים () המדדים של 'ערכים' לאורך 'מימד ההפחתה' של האופרנד 'קלט'. |
סטטי ApproxTopK.Options | isMaxK (בוליאני isMaxK) |
סטטי ApproxTopK.Options | recallTarget (Float recallTarget) |
סטטי ApproxTopK.Options | reductionDimension (Long reductionDimension) |
סטטי ApproxTopK.Options | reductionInputSizeOverride (Long reductionInputSizeOverride) |
פלט <T> | ערכים () ערכי ה-min/max k לאורך ה-'reduction_dimension' של האופרנד 'input'. |
שיטות בירושה
שיטות ציבוריות
סטטי ציבורי ApproxTopK.Options aggregateToTopk (בוליאני aggregateToTopk)
פרמטרים
aggregateToTopk | כאשר זה נכון, מצטבר תוצאות משוערות ל-top-k. כאשר false, מחזיר את התוצאות המשוערות. מספר התוצאות המשוערות מוגדר יישום והוא שווה יותר ל-'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 recallTarget (Float recallTarget)
פרמטרים
recallTarget | יעד זיכרון עבור הקירוב. טווח ב-(0,1] |
---|
public static ApproxTopK.Options reductionDimension (הפחתה ארוכה ממד)
פרמטרים
הפחתה ממד | ממד מספר שלם שלאורכו לחפש. ברירת מחדל: -1. |
---|
public static ApproxTopK.Options reductionInputSizeOverride (Long reductionInputSizeOverride)
פרמטרים
reductionInputSizeOverride | כאשר הוא מוגדר לערך חיובי, הוא עוקף את הגודל שנקבע על ידי `input[reduction_dim]` להערכת ההחזרה. אפשרות זו שימושית כאשר ה'קלט' הנתון הוא רק תת-קבוצה של החישוב הכולל ב-SPMD או צינורות מבוזרים, כאשר לא ניתן לדחות את גודל הקלט האמיתי על ידי צורת ה'קלט'. |
---|
ערכי פלט <T> ציבוריים ()
ערכי ה-min/max k לאורך ה-'reduction_dimension' של האופרנד 'input'. הממד זהה לאופרנד ה-input למעט ה-'reduction_dimension': כאשר 'aggregate_to_topk' נכון, ממד ההפחתה הוא 'k'; אחרת, הוא גדול יותר שווה ל-'k' כאשר הגודל מוגדר במימוש.