CudnnRNNParamsToCanonicalV2

classe final pública CudnnRNNParamsToCanonicalV2

Recupera parâmetros CudnnRNN em forma canônica. Suporta a projeção em LSTM.

Recupera um conjunto de pesos do buffer de parâmetros opaco que pode ser salvo e restaurado de maneira compatível com execuções futuras.

Observe que o buffer de parâmetros pode não ser compatível com diferentes GPUs. Portanto, qualquer salvamento e restauração devem ser convertidos de e para os pesos e preconceitos canônicos.

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. num_params_weights: número da matriz de parâmetros de peso para todas as camadas. num_params_biases: número do vetor de parâmetro de polarização para todas as camadas. pesos: a forma canônica de pesos que pode ser usada para salvar e restaurar. É mais provável que sejam compatíveis entre diferentes gerações. preconceitos: a forma canônica de preconceitos que pode ser usada para salvar e restaurar. É mais provável que sejam compatíveis entre diferentes gerações. 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. num_proj: A dimensionalidade de saída para as matrizes de projeção. Se Nenhum ou 0, nenhuma projeção será executada.

Classes aninhadas

aula CudnnRNNParamsToCanonicalV2.Options Atributos opcionais para CudnnRNNParamsToCanonicalV2

Métodos Públicos

Lista< Saída <T>>
estático <T estende número> CudnnRNNParamsToCanonicalV2 <T>
create (Escopo do escopo , Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Operando <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRNNParamsToCanonicalV2.
CudnnRNNParamsToCanonicalV2.Options estático
direção (direção da string)
CudnnRNNParamsToCanonicalV2.Options estático
abandono (desistência flutuante)
CudnnRNNParamsToCanonicalV2.Options estático
inputMode (String inputMode)
CudnnRNNParamsToCanonicalV2.Options estático
numProj (numProj longo)
CudnnRNNParamsToCanonicalV2.Options estático
rnnMode (String rnnMode)
CudnnRNNParamsToCanonicalV2.Options estático
semente (semente longa)
CudnnRNNParamsToCanonicalV2.Options estático
seed2 (semente2 longa)
Lista< Saída <T>>
pesos ()

Métodos herdados

Métodos Públicos

Lista pública< Saída <T>> preconceitos ()

public static CudnnRNNParamsToCanonicalV2 <T> create ( Escopo escopo , Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Operando <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)

Método de fábrica para criar uma classe que envolve uma nova operação CudnnRNNParamsToCanonicalV2.

Parâmetros
escopo escopo atual
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de CudnnRNNParamsToCanonicalV2

direção pública estática CudnnRNNParamsToCanonicalV2.Options (direção da string)

public static CudnnRNNParamsToCanonicalV2.Options dropout (Float dropout)

público estático CudnnRNNParamsToCanonicalV2.Options inputMode (String inputMode)

público estático CudnnRNNParamsToCanonicalV2.Options numProj (Long numProj)

público estático CudnnRNNParamsToCanonicalV2.Options rnnMode (String rnnMode)

semente pública estática CudnnRNNParamsToCanonicalV2.Options (semente longa)

público estático CudnnRNNParamsToCanonicalV2.Options seed2 (seed2 longo)

Lista pública< Saída <T>> pesos ()