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>> | preconceitos () |
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 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