CudnnRNNBackpropV3

aula final pública CudnnRNNBackpropV3

Etapa de backprop de CudnnRNNV3.

Calcule o backprop de dados e pesos em um RNN. Recebe uma entrada "sequence_lengths" extra que CudnnRNNBackprop.

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. Deve ser "unidirecional" ou "bidirecional". dropout: probabilidade de dropout. 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. input: Se time_major for true, este é um tensor 3-D com a forma de [seq_length, batch_size, input_size]. Se time_major for false, a forma será [batch_size, seq_length, input_size]. input_h: Se time_major for true, este é um tensor 3-D com a forma de [num_layer * dir, batch_size, num_units]. Se time_major for false, a forma será [batch_size, num_layer * dir, num_units]. input_c: Para LSTM, um tensor 3-D com a forma de [num_layer * dir, batch, num_units]. Para outros modelos, é ignorado. params: Um tensor 1-D que contém os pesos e desvios em um layout opaco. O tamanho deve ser criado por meio de CudnnRNNParamsSize e inicializado separadamente. Observe que eles podem não ser compatíveis em diferentes gerações. Portanto, é uma boa ideia salvar e restaurar sequence_lengths: um vetor de comprimentos de cada sequência de entrada. output: Se time_major for true, este é um tensor 3-D com a forma de [seq_length, batch_size, dir * num_units]. Se time_major for false, a forma será [batch_size, seq_length, dir * num_units]. output_h: A mesma forma tem input_h. output_c: A mesma forma que input_c para LSTM. Um tensor vazio para outros modelos. output_backprop: Um tensor 3-D com a mesma forma da saída na passagem para frente. output_h_backprop: Um tensor 3-D com a mesma forma que output_h na passagem direta. output_c_backprop: Um tensor 3-D com a mesma forma que output_c na passagem direta. time_major: Indica se o formato de entrada/saída é maior de tempo ou maior de lote. reserve_space: O mesmo reserve_space produzido na operação de encaminhamento. input_backprop: O backprop a ser inserido na passagem para frente. Tem a mesma forma que a entrada. input_h_backprop: O backprop para input_h na passagem para frente. Tem a mesma forma que input_h. input_c_backprop: O backprop para input_c na passagem para frente. Tem a mesma forma que input_c. params_backprop: O backprop para o buffer de params na passagem de encaminhamento. Tem a mesma forma que params.

Classes aninhadas

aula CudnnRNNBackpropV3.Options Atributos opcionais para CudnnRNNBackpropV3

Métodos públicos

estático <T estende o número> CudnnRNNBackpropV3 <T>
create ( Escopo do escopo, Operando <T> entrada, Operando <T> inputH, Operando <T> inputC, Operando <T> params, Operando <Integer> sequenceLengths, Operando <T> output, Operando <T> outputH, Operando <T > outputC, Operando <T> outputBackprop, Operando <T> outputHBackprop, Operando <T> outputCBackprop, Operando <T> reserveSpace, Operando <?> hostReserved, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRNNBackpropV3.
estático CudnnRNNBackpropV3.Options
direção (direção da string)
estático CudnnRNNBackpropV3.Options
dropout (desistência flutuante)
Saída <T>
Saída <T>
Saída <T>
estático CudnnRNNBackpropV3.Options
inputMode (String inputMode)
estático CudnnRNNBackpropV3.Options
numProj (long numProj)
Saída <T>
estático CudnnRNNBackpropV3.Options
rnnMode (String rnnMode)
estático CudnnRNNBackpropV3.Options
semente (semente longa)
estático CudnnRNNBackpropV3.Options
seed2 (semente longa2)
estático CudnnRNNBackpropV3.Options
timeMajor (Boolean timeMajor)

Métodos Herdados

Métodos públicos

public static CudnnRNNBackpropV3 <T> create ( Escopo do escopo, Operando <T> input, Operando <T> inputH, Operando <T> inputC, Operando <T> params, Operando <Integer> sequenceLengths, Operando <T> output, Operando <T > outputH, Operando <T> outputC, Operando <T> outputBackprop, Operando <T> outputHBackprop, Operando <T> outputCBackprop, Operando <T> reserveSpace, Operando <?> hostReserved, Options... options)

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

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

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

Public static CudnnRNNBackpropV3.Options dropout (dropout float)

Public Output <T> inputBackprop ()

Public Output <T> inputCBackprop ()

Public Output <T> inputHBackprop ()

public static CudnnRNNBackpropV3.Options inputMode (String inputMode)

public static CudnnRNNBackpropV3.Options numProj (Long numProj)

Public Output <T> paramsBackprop ()

public static CudnnRNNBackpropV3.Options rnnMode (String rnnMode)

public static CudnnRNNBackpropV3.Options semente (semente longa)

public static CudnnRNNBackpropV3.Options seed2 (seed2 longo)

public static CudnnRNNBackpropV3.Options timeMajor (Boolean timeMajor)