CudnnRnnParamsSize

Calcula el tamaño de los pesos que puede utilizar un modelo Cudnn RNN.

Devuelve el tamaño de los parámetros que puede utilizar el modelo Cudnn RNN. La asignación de peso y la inicialización posteriores deben utilizar este tamaño.

num_layers: especifica el número de capas en el modelo RNN. num_units: especifica el tamaño del estado oculto. input_size: especifica el tamaño del estado de entrada. rnn_mode: Indica el tipo de modelo RNN. input_mode: indica si hay una proyección lineal entre la entrada y el cálculo real antes de la primera capa. 'skip_input' solo se permite cuando input_size == num_units; 'auto_select' implica 'skip_input' cuando input_size == num_units; de lo contrario, implica 'entrada_lineal'. dirección: Indica si se utilizará un modelo bidireccional. dir = (dirección == bidireccional)? Abandono 2:1: probabilidad de abandono. Cuando se establece en 0, el abandono está deshabilitado. semilla: la primera parte de una semilla para inicializar el abandono. semilla2: la segunda parte de una semilla para inicializar el abandono. params_size: el tamaño del búfer de parámetros que debe asignarse e inicializarse para este modelo RNN. Tenga en cuenta que este búfer de parámetros puede no ser compatible entre GPU. Utilice CudnnRNNParamsWeights y CudnnRNNParamsBiases para guardarlos y restaurarlos de una manera que sea compatible en diferentes ejecuciones.