CudnnRnnParamsSize

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

מחשב את גודל המשקולות שניתן להשתמש בהן על ידי דגם Cudnn RNN.

החזר את גודל הפרמטרים שיכול לשמש את מודל Cudnn RNN. הקצאת משקל ואיתחול לאחר מכן צריכים להשתמש בגודל זה.

num_layers: מציין את מספר השכבות במודל RNN. num_units: מציין את גודל המצב הנסתר. input_size: מציין את גודל מצב הקלט. 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: החלק השני של זרע כדי לאתחל נשירה. params_size: גודל מאגר ה-params שיש להקצות ולאתחל עבור מודל RNN זה. שים לב שייתכן שחוצץ הפרמטרים הזה לא יהיה תואם בין GPUs. אנא השתמש ב- CudnnRNNParamsWeights ו- CudnnRNNParamsBiases כדי לשמור ולשחזר אותם בצורה שתואמת בריצות שונות.

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

מעמד CudnnRnnParamsSize.Options תכונות אופציונליות עבור CudnnRnnParamsSize

קבועים

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

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

פלט <U>
asOutput ()
מחזירה את הידית הסמלית של הטנזור.
סטטי <U מרחיב את TNummer , T מרחיב את Tnumber > CudnnRnnParamsSize <U>
create ( scope scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Class<T> T, Class<U> S, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CudnnRnnParamsSize חדשה.
סטטי CudnnRnnParamsSize.Options
כיוון (כיוון מיתר)
סטטי CudnnRnnParamsSize.Options
נשירה (נשירה ציפה)
סטטי CudnnRnnParamsSize.Options
inputMode (String inputMode)
סטטי CudnnRnnParamsSize.Options
numProj (Long numProj)
פלט <U>
סטטי CudnnRnnParamsSize.Options
rnnMode (מחרוזת rnnMode)
סטטי CudnnRnnParamsSize.Options
זרע (זרע ארוך)
סטטי CudnnRnnParamsSize.Options
seed2 (זרע ארוך2)

שיטות בירושה

קבועים

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

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

ערך קבוע: "CudnnRNNParamsSize"

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

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

מחזירה את הידית הסמלית של הטנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

public static CudnnRnnParamsSize <U> create ( היקף scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Class<T> T, Class<U> S, Options... options)

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

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

סטטי ציבורי CudnnRnnParamsSize.Options כיוון (כיוון מחרוזת)

ציבורי סטטי CudnnRnnParamsSize.Options נשירה (נשירה ציפה)

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

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

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

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

סטטי ציבורי CudnnRnnParamsSize.Options seed (סיד ארוך)

סטטי ציבורי CudnnRnnParamsSize.Options seed2 (גרעין ארוך2)