CudnnRNNCanonicalToParamsV2

classe final pública CudnnRNNCanonicalToParamsV2

Converte os 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 ​​no próximo treinamento ou inferências.

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

Métodos públicos

Saída <T>
comoSaída ()
Retorna o identificador simbólico de um tensor.
estático <T extend Number> CudnnRNNCanonicalToParamsV2 <T>
create ( Escopo do escopo, Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Iterable< Operando <T>> pesos, Iterable< Operando <T>> bias, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRNNCanonicalToParamsV2.
static CudnnRNNCanonicalToParamsV2.Options
direção (direção da string)
static CudnnRNNCanonicalToParamsV2.Options
dropout (desistência flutuante)
static CudnnRNNCanonicalToParamsV2.Options
inputMode (String inputMode)
static CudnnRNNCanonicalToParamsV2.Options
numProj (long numProj)
Saída <T>
static CudnnRNNCanonicalToParamsV2.Options
rnnMode (String rnnMode)
static CudnnRNNCanonicalToParamsV2.Options
semente (semente longa)
static CudnnRNNCanonicalToParamsV2.Options
seed2 (semente longa2)

Métodos Herdados

Métodos públicos

Public Output <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 do escopo, Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Iterable< Operando <T>> pesos, Iterable< Operando <T>> bias, Options.. . opções)

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

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

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

public static CudnnRNNCanonicalToParamsV2.Options dropout (Float dropout)

public static CudnnRNNCanonicalToParamsV2.Options inputMode (String inputMode)

public static CudnnRNNCanonicalToParamsV2.Options numProj (Long numProj)

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

public static CudnnRNNCanonicalToParamsV2.Options rnnMode (String rnnMode)

public static CudnnRNNCanonicalToParamsV2.Options semente (semente longa)

public static CudnnRNNCanonicalToParamsV2.Options seed2 (Long seed2)