מחשב את גודל המשקולות שניתן להשתמש בהן על ידי דגם Cudnn RNN.
החזר את גודל הפרמטרים שיכול לשמש את מודל Cudnn RNN. הקצאת משקל ואיתחול לאחר מכן צריכים להשתמש בגודל זה.
num_layers: מציין את מספר השכבות במודל RNN. num_units: מציין את גודל המצב הנסתר. input_size: מציין את גודל מצב הקלט. rnn_mode: מציין את סוג דגם ה-RNN. input_mode: ציין אם יש השלכה ליניארית בין הקלט לבין החישוב בפועל לפני השכבה הראשונה. 'skip_input' מותר רק כאשר input_size == num_units; 'auto_select' מרמז על 'skip_input' כאשר input_size == num_units; אחרת, זה מרמז על 'קלט_ליניארי'. כיוון: מציין אם ייעשה שימוש במודל דו-כיווני. dir = (כיוון == דו כיווני) ? 2:1 נשירה: הסתברות לנשירה. כאשר מוגדר ל-0., נשירה מושבתת. seed: החלק הראשון של זרע כדי לאתחל נשירה. seed2: החלק השני של זרע כדי לאתחל נשירה. params_size: גודל מאגר ה-params שיש להקצות ולאתחל עבור מודל RNN זה. שים לב שייתכן שחוצץ הפרמטרים הזה לא יהיה תואם בין GPUs. אנא השתמש ב- CudnnRNNParamsWeights ו- CudnnRNNParamsBiases כדי לשמור ולשחזר אותם בצורה שתואמת בריצות שונות.
כיתות מקוננות
מעמד | CudnnRnnParamsSize.Options | תכונות אופציונליות עבור CudnnRnnParamsSize |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
פלט <U> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
סטטי <U מרחיב את TNummer , T מרחיב את Tnumber > CudnnRnnParamsSize <U> | |
סטטי CudnnRnnParamsSize.Options | כיוון (כיוון מיתר) |
סטטי CudnnRnnParamsSize.Options | נשירה (נשירה ציפה) |
סטטי CudnnRnnParamsSize.Options | inputMode (String inputMode) |
סטטי CudnnRnnParamsSize.Options | numProj (Long numProj) |
פלט <U> | paramsSize () |
סטטי CudnnRnnParamsSize.Options | rnnMode (מחרוזת rnnMode) |
סטטי CudnnRnnParamsSize.Options | זרע (זרע ארוך) |
סטטי CudnnRnnParamsSize.Options | seed2 (זרע ארוך2) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
פלט ציבורי <U> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
public static CudnnRnnParamsSize <U> create ( היקף scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Class<T> T, Class<U> S, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRnnParamsSize חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של CudnnRnnParamsSize