CudnnRnnParamsSize

Calcula o tamanho dos pesos que podem ser usados ​​por um modelo Cudnn RNN.

Retorne o tamanho dos parâmetros que pode ser usado pelo modelo Cudnn RNN. A alocação de peso subsequente e a inicialização devem usar esse tamanho.

num_layers: especifica o número de camadas no modelo RNN. num_units: especifica o tamanho do estado oculto. input_size: especifica o tamanho do estado de entrada. rnn_mode: indica o tipo do modelo RNN. input_mode: indica se há uma projeção linear entre a entrada e o cálculo real antes da primeira camada. 'skip_input' só é permitido quando input_size == num_units; 'auto_select' implica 'skip_input' quando input_size == num_units; caso contrário, implica em 'entrada_linear'. direção: indica se um modelo bidirecional será usado. dir = (direção == bidirecional)? 2: 1 abandono: probabilidade de abandono. Quando definido como 0., o dropout é desabilitado. semente: a 1ª parte de uma semente para inicializar o dropout. seed2: a 2ª parte de um seed para inicializar o dropout. params_size: O tamanho do buffer params que deve ser alocado e inicializado para este modelo RNN. Observe que este buffer de parâmetros pode não ser compatível com GPUs. Use CudnnRNNParamsWeights e CudnnRNNParamsBiases para salvá-los e restaurá-los de uma maneira que seja compatível em diferentes execuções.