CudnnRNNParamsToCanonical

classe final pública CudnnRNNParamsToCanonical

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) ? 2: 1 abandono: probabilidade de abandono. 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 CudnnRNNParamsToCanonical.Options Atributos opcionais para CudnnRNNParamsToCanonical

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

Lista< Saída <T>>
estático <T estende TNumber > CudnnRNNParamsToCanonical <T>
create ( Escopo do escopo , Operando < TInt32 > numLayers, Operando < TInt32 > numUnits, Operando < TInt32 > 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 CudnnRNNParamsToCanonical.
estático CudnnRNNParamsToCanonical.Options
direção (direção da string)
estático CudnnRNNParamsToCanonical.Options
abandono (desistência flutuante)
estático CudnnRNNParamsToCanonical.Options
inputMode (String inputMode)
estático CudnnRNNParamsToCanonical.Options
numProj (numProj longo)
estático CudnnRNNParamsToCanonical.Options
rnnMode (String rnnMode)
estático CudnnRNNParamsToCanonical.Options
semente (semente longa)
estático CudnnRNNParamsToCanonical.Options
seed2 (semente2 longa)
Lista< Saída <T>>
pesos ()

Métodos herdados

Constantes

String final estática pública OP_NAME

O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Valor constante: "CudnnRNNParamsToCanonicalV2"

Métodos Públicos

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

public static CudnnRNNParamsToCanonical <T> create ( Escopo de escopo , Operando < TInt32 > numLayers, Operando < TInt32 > numUnits, Operando < TInt32 > inputSize, Operando <T> params, Long numParamsWeights, Long numParamsBiases, Opções... opções)

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

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

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

public static CudnnRNNParamsToCanonical.Options dropout (Float dropout)

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

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

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

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

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

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