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