Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

CudnnRNNV3

CudnnRNNV3 classe final pública

Um RNN apoiado por cuDNN.

Calcula o RNN da entrada e dos estados iniciais, em relação ao buffer de parâmetros. Aceita uma entrada extra "sequence_lengths" do que CudnnRNN.

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. Deve ser "unidirecional" ou "bidirecional". abandono: probabilidade de abandono. Quando definido como 0., o dropout é desabilitado. seed: A primeira parte de um seed para inicializar o dropout. seed2: A segunda parte de um seed para inicializar o dropout. entrada: Se time_major for true, este é um tensor 3-D com a forma de [comprimento_seq, tamanho_do_lote, tamanho_de_entrada]. 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, ele é ignorado. params: Um tensor 1-D que contém os pesos e vieses 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. saída: Se time_major for true, este é um tensor 3-D com a forma de [comprimento_seq, tamanho_do_lote, dir * unidades_um]. 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: O mesmo formato que input_c para LSTM. Um tensor vazio para outros modelos. is_training: indica se esta operação é usada para inferência ou treinamento. time_major: indica se o formato de entrada / saída é principal de tempo ou principal de lote. reserve_space: Um tensor opaco que pode ser usado no cálculo de backprop. Ele só é produzido se is_training for true.

Classes aninhadas

classe CudnnRNNV3.Options Atributos opcionais para CudnnRNNV3

Métodos Públicos

estática <T estende Number> CudnnRNNV3 <T>
criar ( Scope escopo, Operando <T> entrada, Operando <T> inputH, Operando <T> inputC, Operando <T> params, Operando <Integer> sequenceLengths, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação CudnnRNNV3.
estáticos CudnnRNNV3.Options
direcção (direcção da corda)
estáticos CudnnRNNV3.Options
abandono (abandono do flutuador)
Output <?>
estáticos CudnnRNNV3.Options
inputMode (String inputMode)
estáticos CudnnRNNV3.Options
isTraining (booleano isTraining)
estáticos CudnnRNNV3.Options
numProj (Long numProj)
Output <T>
saída ()
Output <T>
Output <T>
Output <T>
estáticos CudnnRNNV3.Options
rnnMode (String rnnMode)
estáticos CudnnRNNV3.Options
sementes (semente Long)
estáticos CudnnRNNV3.Options
seed2 (seed2 Long)
estáticos CudnnRNNV3.Options
timeMajor (booleano timeMajor)

Métodos herdados

Métodos Públicos

public static CudnnRNNV3 <T> create ( Scope escopo, Operando <T> entrada, Operando <T> inputH, Operando <T> inputC, Operando <T> params, Operando <Integer> sequenceLengths, Options ... Opções)

Método de fábrica para criar uma classe envolvendo uma nova operação CudnnRNNV3.

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

public static CudnnRNNV3.Options direção (sentido String)

public static CudnnRNNV3.Options abandono (abandono Float)

pública de saída <?> hostReserved ()

public static CudnnRNNV3.Options inputMode (String inputMode)

public static CudnnRNNV3.Options isTraining (Boolean isTraining)

public static CudnnRNNV3.Options numProj (Long numProj)

pública Output <T> de saída ()

pública Output <T> outputC ()

pública Output <T> outputH ()

pública Output <T> reserveSpace ()

public static CudnnRNNV3.Options rnnMode (String rnnMode)

public static CudnnRNNV3.Options semente (seed Long)

public static CudnnRNNV3.Options seed2 (seed2 Long)

public static CudnnRNNV3.Options timeMajor (Boolean timeMajor)