Calcula o tamanho dos pesos que podem ser usados por um modelo Cudnn RNN.
Retorne o tamanho dos parâmetros que podem ser usados pelo modelo Cudnn RNN. A alocação de peso e a inicialização subsequentes 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 existe 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 'linear_input'. direção: Indica se um modelo bidirecional será usado. dir = (direção == bidirecional) ? Desistência 2: 1: probabilidade de desistência. Quando definido como 0, o dropout é desabilitado. seed: a 1ª parte de uma seed para inicializar o dropout. seed2: a 2ª parte de uma semente para inicializar o dropout. params_size: O tamanho do buffer de parâmetros que deve ser alocado e inicializado para este modelo RNN. Observe que esse buffer de parâmetros pode não ser compatível entre GPUs. Use CudnnRNNParamsWeights e CudnnRNNParamsBiases para salvá-los e restaurá-los de uma forma que seja compatível em diferentes execuções.
Classes aninhadas
aula | CudnnRnnParamsSize.Options | Atributos opcionais para CudnnRnnParamsSize |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <U> | asOutput () Retorna o identificador simbólico do tensor. |
estático <U estende TNumber , T estende TNumber > CudnnRnnParamsSize <U> | |
CudnnRnnParamsSize.Options estático | direção (direção da string) |
CudnnRnnParamsSize.Options estático | abandono (desistência flutuante) |
CudnnRnnParamsSize.Options estático | inputMode (String inputMode) |
CudnnRnnParamsSize.Options estático | numProj (numProj longo) |
Saída <U> | paramSize () |
CudnnRnnParamsSize.Options estático | rnnMode (String rnnMode) |
CudnnRnnParamsSize.Options estático | semente (semente longa) |
CudnnRnnParamsSize.Options estático | seed2 (semente2 longa) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <U> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static CudnnRnnParamsSize <U> create ( Escopo escopo , Operando < TInt32 > numLayers, Operando < TInt32 > numUnits, Operando < TInt32 > inputSize, Classe<T> T, Classe<U> S, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRnnParamsSize.
Parâmetros
escopo | escopo atual |
---|---|
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de CudnnRnnParamsSize