CudnnRNNCanonicalToParamsV2

classe final pública CudnnRNNCanonicalToParamsV2

Converte parâmetros CudnnRNN da forma canônica para a forma utilizável. Suporta a projeção em LSTM.

Grava um conjunto de pesos no buffer de parâmetros opacos para que possam ser usados ​​em treinamentos ou inferências futuros.

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. 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. 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. 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 CudnnRNNCanonicalToParamsV2.Options Atributos opcionais para CudnnRNNCanonicalToParamsV2

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estático <T estende número> CudnnRNNCanonicalToParamsV2 <T>
create ( Escopo do escopo , Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Iterable< Operando <T>> pesos, Iterable< Operando <T>> preconceitos, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRNNCanonicalToParamsV2.
estático CudnnRNNCanonicalToParamsV2.Options
direção (direção da string)
estático CudnnRNNCanonicalToParamsV2.Options
abandono (desistência flutuante)
estático CudnnRNNCanonicalToParamsV2.Options
inputMode (String inputMode)
estático CudnnRNNCanonicalToParamsV2.Options
numProj (numProj longo)
Saída <T>
estático CudnnRNNCanonicalToParamsV2.Options
rnnMode (String rnnMode)
estático CudnnRNNCanonicalToParamsV2.Options
semente (semente longa)
estático CudnnRNNCanonicalToParamsV2.Options
seed2 (semente2 longa)

Métodos herdados

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static CudnnRNNCanonicalToParamsV2 <T> create ( Escopo de escopo , Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Iterable< Operando <T>> pesos, Iterable< Operando <T>> preconceitos, Opções.. .opções )

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

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

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

public static CudnnRNNCanonicalToParamsV2.Options dropout (Float dropout)

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

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

Saída pública <T> parâmetros ()

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

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

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