מאחזר פרמטרים של 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: ממדיות הפלט עבור מטריצות ההקרנה. אם אין או 0, לא מתבצעת הקרנה.
כיתות מקוננות
מעמד | CudnnRNNParamsToCanonical.Options | תכונות אופציונליות עבור CudnnRNNParamsToCanonical |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
רשימה< פלט <T>> | הטיות () |
סטטי <T מרחיב TNummer > CudnnRNNParamsToCanonical <T> | |
סטטי CudnnRNNParamsToCanonical.Options | כיוון (כיוון מיתר) |
סטטי CudnnRNNParamsToCanonical.Options | נשירה (נשירה ציפה) |
סטטי CudnnRNNParamsToCanonical.Options | inputMode (String inputMode) |
סטטי CudnnRNNParamsToCanonical.Options | numProj (Long numProj) |
סטטי CudnnRNNParamsToCanonical.Options | rnnMode (מחרוזת rnnMode) |
סטטי CudnnRNNParamsToCanonical.Options | זרע (זרע ארוך) |
סטטי CudnnRNNParamsToCanonical.Options | seed2 (זרע ארוך2) |
רשימה< פלט <T>> | משקולות () |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
public static CudnnRNNParamsToCanonical <T> create ( scope scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Operand <T> פרמטרים, Long numParamsWeights, Long numParamsBias ... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולה חדשה של CudnnRNNParamsToCanonical.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של CudnnRNNParamsToCanonical