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> | parâmetros () |
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