CudnnRNNCanonicalToParamsV2

public final class 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 futuros treinamentos ou inferências.

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

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
static <T extends Number> CudnnRNNCanonicalToParamsV2 <T>
criar ( Scope escopo, Operando <inteiro> numLayers, Operando <inteiro> numUnits, Operando <Integer> inputSize, Iterable < Operando <T >> pesos, Iterable < Operando <T >> preconceitos, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação CudnnRNNCanonicalToParamsV2.
static CudnnRNNCanonicalToParamsV2.Options
static CudnnRNNCanonicalToParamsV2.Options
dropout (flutuação dropout)
static CudnnRNNCanonicalToParamsV2.Options
inputMode (String inputMode)
static CudnnRNNCanonicalToParamsV2.Options
numProj (Long numProj)
Saída <T>
params ()
static CudnnRNNCanonicalToParamsV2.Options
rnnMode (String rnnMode)
static CudnnRNNCanonicalToParamsV2.Options
semente ( semente longa)
static CudnnRNNCanonicalToParamsV2.Options
seed2 ( semente2 longa)

Métodos herdados

Métodos Públicos

public Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as 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 ( Scope 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 envolvendo 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

public static CudnnRNNCanonicalToParamsV2.Options direction (String direction)

public static CudnnRNNCanonicalToParamsV2.Options dropout (Float dropout)

public static CudnnRNNCanonicalToParamsV2.Options inputMode (String inputMode)

public static CudnnRNNCanonicalToParamsV2.Options numProj (Long numProj)

public Output <T> params ()

public static CudnnRNNCanonicalToParamsV2.Options rnnMode (String rnnMode)

public static CudnnRNNCanonicalToParamsV2.Options seed (Long seed)

public static CudnnRNNCanonicalToParamsV2.Options seed2 (Long seed2)