CudnnRNN

כיתת גמר ציבורית CudnnRNN

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 נכון.

כיתות מקוננות

מעמד CudnnRNN.אפשרויות תכונות אופציונליות עבור CudnnRNN

קבועים

חוּט OP_NAME השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

שיטות ציבוריות

סטטי <T מרחיב TNummer > CudnnRNN <T>
create ( scope scope, Operand <T> input, Operand <T> inputH, Operand <T> inputC, Operand <T> params, Operand < TInt32 > sequenceLengths, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRNN חדשה.
סטטי CudnnRNN.Options
כיוון (כיוון מיתר)
סטטי CudnnRNN.Options
נשירה (נשירה ציפה)
פלט <?>
סטטי CudnnRNN.Options
inputMode (String inputMode)
סטטי CudnnRNN.Options
isTraining (Boolian isTraining)
סטטי CudnnRNN.Options
numProj (Long numProj)
פלט <T>
פלט ()
פלט <T>
פלט <T>
פלט <T>
סטטי CudnnRNN.Options
rnnMode (מחרוזת rnnMode)
סטטי CudnnRNN.Options
זרע (זרע ארוך)
סטטי CudnnRNN.Options
seed2 (זרע ארוך2)
סטטי CudnnRNN.Options
timeMajor (זמן בוליאני)

שיטות בירושה

קבועים

מחרוזת סופית סטטית ציבורית OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

ערך קבוע: "CudnnRNNV3"

שיטות ציבוריות

ציבורי סטטי CudnnRNN <T> create ( היקף היקף , קלט Operand <T>, קלט Operand <T> inputH, Operand <T> inputC, Params Operand <T>, Operand < TInt32 > רצף אורכים, אפשרויות... אפשרויות)

שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRNN חדשה.

פרמטרים
תְחוּם ההיקף הנוכחי
אפשרויות נושא ערכי תכונות אופציונליות
החזרות
  • מופע חדש של CudnnRNN

ציבורי סטטי CudnnRNN. כיוון אפשרויות (כיוון מחרוזת)

ציבורי סטטי CudnnRNN. נשירה של אפשרויות (נשירה ציפה)

פלט ציבורי <?> hostReserved ()

ציבורי סטטי CudnnRNN.Options inputMode (String inputMode)

ציבורי סטטי CudnnRNN.Options isTraining (Boolean isTraining)

סטטי ציבורי CudnnRNN.Options numProj (Long numProj)

פלט ציבורי <T> פלט ()

פלט ציבורי <T> פלטC ()

פלט ציבורי <T> פלטH ()

פלט ציבורי <T> reserveSpace ()

ציבורי סטטי CudnnRNN.Options rnnMode (String rnnMode)

ציבורי סטטי CudnnRNN. Options seed (Long seed)

ציבורי סטטי CudnnRNN.Options seed2 (Long seed2)

ציבורי סטטי CudnnRNN.Options timeMajor (זמן בוליאני Major)