CudnnRNNParamsToCanonicalV2

CudnnRNNParamsToCanonicalV2 classe final pública

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

Recupera um conjunto de pesos do buffer de parâmetros opacos que podem ser salvos e restaurados de forma 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 vieses 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âmetro 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 podem ser usados ​​para salvar e restaurar. É mais provável que sejam compatíveis entre gerações diferentes. vieses: a forma canônica de vieses que pode ser usada para salvar e restaurar. Eles são mais propensos a serem compatíveis entre gerações diferentes. 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. num_proj: a dimensionalidade de saída para as matrizes de projeção. Se Nenhum ou 0, nenhuma projeção é executada.

Classes aninhadas

classe CudnnRNNParamsToCanonicalV2.Options Atributos opcionais para CudnnRNNParamsToCanonicalV2

Métodos Públicos

List < Output <T >>
estática <T estende Number> CudnnRNNParamsToCanonicalV2 <T>
criar ( Scope escopo, Operando <inteiro> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Operando <T> params, numParamsWeights longo, longo numParamsBiases, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação CudnnRNNParamsToCanonicalV2.
estáticos CudnnRNNParamsToCanonicalV2.Options
direcção (direcção da corda)
estáticos CudnnRNNParamsToCanonicalV2.Options
abandono (abandono do flutuador)
estáticos CudnnRNNParamsToCanonicalV2.Options
inputMode (String inputMode)
estáticos CudnnRNNParamsToCanonicalV2.Options
numProj (Long numProj)
estáticos CudnnRNNParamsToCanonicalV2.Options
rnnMode (String rnnMode)
estáticos CudnnRNNParamsToCanonicalV2.Options
sementes (semente Long)
estáticos CudnnRNNParamsToCanonicalV2.Options
seed2 (seed2 Long)
List < Output <T >>
pesos ()

Métodos herdados

Métodos Públicos

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

public static CudnnRNNParamsToCanonicalV2 <T> criar ( Scope escopo, Operando <inteiro> numLayers, Operando <inteiro> numUnits, Operando <Integer> inputSize, Operando <T> params, numParamsWeights longo, longo numParamsBiases, Options ... Opções)

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

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

public static CudnnRNNParamsToCanonicalV2.Options direção (sentido String)

public static CudnnRNNParamsToCanonicalV2.Options abandono (abandono Float)

public static CudnnRNNParamsToCanonicalV2.Options inputMode (String inputMode)

public static CudnnRNNParamsToCanonicalV2.Options numProj (Long numProj)

public static CudnnRNNParamsToCanonicalV2.Options rnnMode (String rnnMode)

public static CudnnRNNParamsToCanonicalV2.Options semente (seed Long)

public static CudnnRNNParamsToCanonicalV2.Options seed2 (seed2 Long)

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