CudnnRNNParamsToCanonicalV2

public final class CudnnRNNParamsToCanonicalV2

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. É mais provável que sejam 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

aula CudnnRNNParamsToCanonicalV2.Options Atributos opcionais para CudnnRNNParamsToCanonicalV2

Métodos Públicos

Listar < Output <T>>
static <T extends 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.
static CudnnRNNParamsToCanonicalV2.Options
static CudnnRNNParamsToCanonicalV2.Options
dropout (flutuação dropout)
static CudnnRNNParamsToCanonicalV2.Options
inputMode (String inputMode)
static CudnnRNNParamsToCanonicalV2.Options
numProj (Long numProj)
static CudnnRNNParamsToCanonicalV2.Options
rnnMode (String rnnMode)
static CudnnRNNParamsToCanonicalV2.Options
semente ( semente longa)
static CudnnRNNParamsToCanonicalV2.Options
seed2 ( semente2 longa)
Listar < Output <T>>
pesos ()

Métodos herdados

Métodos Públicos

Lista pública < Output <T>> enviesamentos ()

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 direction (String direction)

public static CudnnRNNParamsToCanonicalV2.Options dropout (Float dropout)

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 seed (Long seed)

public static CudnnRNNParamsToCanonicalV2.Options seed2 (Long seed2)

Pesos public List < Output <T>> ()