RNN מגובה על ידי cuDNN.
מחשב את ה-RNN ממצבי הקלט וההתחלה, ביחס למאגר הפרמטרים. מקבל קלט אחד נוסף "sequence_lengths" מאשר CudnnRNN.
rnn_mode: מציין את סוג דגם ה-RNN. input_mode: מציין אם יש השלכה ליניארית בין הקלט לחישוב בפועל לפני השכבה הראשונה. 'skip_input' מותר רק כאשר input_size == num_units; 'auto_select' מרמז על 'skip_input' כאשר input_size == num_units; אחרת, זה מרמז על 'קלט_ליניארי'. כיוון: מציין אם ייעשה שימוש במודל דו-כיווני. צריך להיות "חד-כיווני" או "דו-כיווני". נשירה: הסתברות נשירה. כאשר מוגדר ל-0., נשירה מושבתת. seed: החלק הראשון של זרע כדי לאתחל נשירה. seed2: החלק השני של זרע כדי לאתחל נשירה. קלט: אם time_major נכון, זהו טנזור תלת-ממדי עם הצורה של [seq_length, batch_size, input_size]. אם time_major הוא false, הצורה היא [batch_size, seq_length, input_size]. input_h: אם time_major נכון, זהו טנזור תלת-ממדי עם הצורה של [num_layer * dir, batch_size, num_units]. אם time_major הוא false, הצורה היא [אצווה_גודל, num_layer * dir, num_units]. input_c: עבור LSTM, טנזור תלת-ממדי עם הצורה של [num_layer * dir, batch, num_units]. עבור דגמים אחרים, זה מתעלם. params: טנזור 1-D המכיל את המשקולות וההטיות בפריסה אטומה. יש ליצור את הגודל דרך CudnnRNNParamsSize, ולאתחל בנפרד. שים לב שייתכן שהם לא תואמים לדורות שונים. אז זה רעיון טוב לשמור ולשחזר sequence_lengths: וקטור של אורכים של כל רצף קלט. פלט: אם time_major נכון, זהו טנזור תלת-ממדי עם הצורה של [seq_length, batch_size, dir * num_units]. אם time_major הוא false, הצורה היא [batch_size, seq_length, dir * num_units]. output_h: לאותה צורה יש input_h. output_c: אותה צורה כמו input_c עבור LSTM. טנסור ריק עבור דגמים אחרים. is_training: מציין אם הפעולה הזו משמשת להסקת הסקה או אימון. time_major: מציין אם פורמט הקלט/פלט הוא זמן עיקרי או אצווה עיקרי. reserve_space: טנסור אטום שניתן להשתמש בו בחישוב ה-backprop. זה מופק רק אם is_training נכון.
כיתות מקוננות
| מעמד | CudnnRNNV3.Options | תכונות אופציונליות עבור CudnnRNNV3 | |
שיטות ציבוריות
| סטטי <T מרחיב מספר> CudnnRNNV3 <T> | |
| סטטי CudnnRNNV3.Options | כיוון (כיוון מיתר) |
| סטטי CudnnRNNV3.Options | נשירה (נשירה ציפה) |
| פלט <?> | מארח שמור () |
| סטטי CudnnRNNV3.Options | inputMode (String inputMode) |
| סטטי CudnnRNNV3.Options | isTraining (Boolian isTraining) |
| סטטי CudnnRNNV3.Options | numProj (Long numProj) |
| פלט <T> | פלט () |
| פלט <T> | outputC () |
| פלט <T> | פלטH () |
| פלט <T> | ReserveSpace () |
| סטטי CudnnRNNV3.Options | rnnMode (מחרוזת rnnMode) |
| סטטי CudnnRNNV3.Options | זרע (זרע ארוך) |
| סטטי CudnnRNNV3.Options | seed2 (זרע ארוך2) |
| סטטי CudnnRNNV3.Options | timeMajor (זמן בוליאני) |
שיטות בירושה
שיטות ציבוריות
סטטי ציבורי CudnnRNNV3 <T> create ( היקף היקף , קלט Operand <T>, קלט Operand <T> inputH, Operand <T> inputC, פרמטרים של Operand <T>, רצף אורכי אופרנד <Integer>, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRNNV3 חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של CudnnRNNV3