CudnnRNNParamsToCanonicalV2

classe final pública CudnnRNNParamsToCanonicalV2

Recupera os 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 em 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âmetros de peso para todas as camadas. num_params_biases: número de vetores de parâmetros de bias para todas as camadas. pesos: a forma canônica de pesos que pode ser usada para salvar e restaurar. Eles são mais propensos a serem compatíveis em diferentes gerações. vieses: a forma canônica de vieses que pode ser usada para salvar e restaurar. Eles são mais propensos a serem compatíveis em diferentes gerações. rnn_mode: Indica o tipo do modelo RNN. input_mode: Indica se existe uma projeção linear entre a entrada e a computação 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 será utilizado um modelo bidirecional. dir = (direção == bidirecional) ? 2 : 1 desistência: 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 é 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.
static CudnnRNNParamsToCanonicalV2.Options
direção (direção da string)
static CudnnRNNParamsToCanonicalV2.Options
dropout (desistência flutuante)
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 (semente longa2)
Lista< Saída <T>>
pesos ()

Métodos Herdados

Métodos públicos

Public List< Output <T>> bias ()

public static 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.

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

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

public static CudnnRNNParamsToCanonicalV2.Options dropout (desistência flutuante)

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 (semente longa)

public static CudnnRNNParamsToCanonicalV2.Options seed2 (Long seed2)

public List< Saída <T>> pesos ()