ממיר פרמטרים של CudnnRNN מצורה קנונית לצורה שמישה. זה תומך בהקרנה ב-LSTM.
כותב קבוצה של משקולות למאגר הפרמטרים האטומים כדי שיוכלו לשמש באימונים הקרובים או במסקנות.
שים לב שייתכן שמאגר הפרמטרים לא יהיה תואם בין מעבדי GPU שונים. אז יש להמיר כל שמירה ושיקום למשקלים ולהטיות הקנוניות.
num_layers: מציין את מספר השכבות במודל RNN. num_units: מציין את גודל המצב הנסתר. input_size: מציין את גודל מצב הקלט. משקולות: הצורה הקנונית של משקולות שניתן להשתמש בהן לשמירה ושיקום. סביר יותר שהם יהיו תואמים על פני דורות שונים. הטיות: הצורה הקנונית של הטיות שניתן להשתמש בהן לשמירה ושיקום. סביר יותר שהם יהיו תואמים על פני דורות שונים. num_params_weights: מספר מטריצת פרמטר המשקל עבור כל השכבות. num_params_biases: מספר וקטור פרמטר הטיה עבור כל השכבות. 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: החלק השני של זרע כדי לאתחל נשירה. num_proj: ממדיות הפלט עבור מטריצות ההקרנה. אם None או 0, לא מתבצעת הקרנה.
כיתות מקוננות
מעמד | CudnnRNNCanonicalToParams.Options | תכונות אופציונליות עבור CudnnRNNCanonicalToParams |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
סטטי <T מרחיב את TNummer > CudnnRNNCanonicalToParams <T> | |
סטטי CudnnRNNCanonicalToParams.Options | כיוון (כיוון מיתר) |
סטטי CudnnRNNCanonicalToParams.Options | נשירה (נשירה ציפה) |
סטטי CudnnRNNCanonicalToParams.Options | inputMode (String inputMode) |
סטטי CudnnRNNCanonicalToParams.Options | numProj (Long numProj) |
פלט <T> | פרמטרים () |
סטטי CudnnRNNCanonicalToParams.Options | rnnMode (מחרוזת rnnMode) |
סטטי CudnnRNNCanonicalToParams.Options | זרע (זרע ארוך) |
סטטי CudnnRNNCanonicalToParams.Options | seed2 (זרע ארוך2) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
Public static CudnnRNNCanonicalToParams <T> create ( היקף scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Iterable< Operand <T>> משקלים, הטיות Iterable< Operand <T>> ., . אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRNNCanonicalToParams חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של CudnnRNNCanonicalToParams