اندازه وزن هایی را محاسبه می کند که می تواند توسط یک مدل 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 تخصیص داده شود و مقداردهی اولیه شود. توجه داشته باشید که این بافر پارامترها ممکن است با پردازندههای گرافیکی سازگار نباشد. لطفاً از CudnnRNNParamsWeights و CudnnRNNParamsBiases برای ذخیره و بازیابی آنها به روشی که در اجراهای مختلف سازگار باشد، استفاده کنید.
کلاس های تو در تو
کلاس | CudnnRnnParamsSize.Options | ویژگی های اختیاری برای CudnnRnnParamsSize |
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
خروجی <U> | asOutput () دسته نمادین تانسور را برمی گرداند. |
static <U گسترش TNumber ، T گسترش TNumber > CudnnRnnParamsSize <U> | |
استاتیک CudnnRnnParamsSize.Options | جهت (جهت رشته) |
استاتیک CudnnRnnParamsSize.Options | انصراف (ترک تحصیل شناور) |
استاتیک CudnnRnnParamsSize.Options | حالت ورودی (حالت ورودی رشته) |
استاتیک CudnnRnnParamsSize.Options | numProj (NumProj طولانی) |
خروجی <U> | paramsSize () |
استاتیک CudnnRnnParamsSize.Options | rnnMode (رشته rnnMode) |
استاتیک CudnnRnnParamsSize.Options | دانه (دانه بلند) |
استاتیک CudnnRnnParamsSize.Options | دانه 2 (دانه بلند 2) |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
خروجی عمومی <U> asOutput ()
دسته نمادین تانسور را برمی گرداند.
ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.
استاتیک عمومی CudnnRnnParamsSize <U> ایجاد ( دامنه دامنه ، عملوند < TINT32 > numLayers، عملوند < TINT32 > numUnits، عملوند < TINT32 > inputSize، کلاس<T> T، کلاس<U> S، گزینهها ... )
روش کارخانه برای ایجاد کلاسی که یک عملیات CudnnRnnParamsSize جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از CudnnRnnParamsSize