CudnnRnnParamsSize

Calcula el tamaño de las ponderaciones que puede utilizar un modelo Cudnn RNN.

Devuelve el tamaño de 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 'linear_input'. direction: Indica si se utilizará un modelo bidireccional. dir = (dirección == bidireccional)? Abandono 2: 1: probabilidad de abandono. Cuando se establece en 0., la deserción está deshabilitada. semilla: la primera parte de una semilla para inicializar la deserción. seed2: la segunda parte de una semilla para inicializar la deserción. params_size: el tamaño del búfer params que debe asignarse e inicializarse para este modelo RNN. Tenga en cuenta que es posible que este búfer de parámetros no sea compatible con todas las GPU. Utilice CudnnRNNParamsWeights y CudnnRNNParamsBiases para guardarlos y restaurarlos de una manera que sea compatible en diferentes ejecuciones.