CudnnRNNBackpropV3

public final class CudnnRNNBackpropV3

Step backprop di CudnnRNNV3.

Calcola il backprop di dati e pesi in un RNN. Accetta un input "sequence_lengths" aggiuntivo rispetto a CudnnRNNBackprop.

rnn_mode: indica il tipo di modello RNN. input_mode: indica se esiste una proiezione lineare tra l'input e il calcolo effettivo prima del primo livello. 'skip_input' è consentito solo quando input_size == num_units; "auto_select" implica "skip_input" quando input_size == num_units; in caso contrario, implica "linear_input". direction: indica se verrà utilizzato un modello bidirezionale. Dovrebbe essere "unidirezionale" o "bidirezionale". abbandono: probabilità di abbandono. Quando è impostato su 0., il dropout è disabilitato. seed: la prima parte di un seme per inizializzare l'abbandono. seed2: la seconda parte di un seme per inizializzare l'abbandono. input: se time_major è true, questo è un tensore 3-D con la forma di [seq_length, batch_size, input_size]. Se time_major è false, la forma è [batch_size, seq_length, input_size]. input_h: se time_major è vero, questo è un tensore 3-D con la forma di [num_layer * dir, batch_size, num_units]. Se time_major è false, la forma è [batch_size, num_layer * dir, num_units]. input_c: per LSTM, un tensore 3-D con la forma di [num_layer * dir, batch, num_units]. Per altri modelli, viene ignorato. params: un tensore 1-D che contiene i pesi e i bias in un layout opaco. La dimensione deve essere creata tramite CudnnRNNParamsSize e inizializzata separatamente. Tieni presente che potrebbero non essere compatibili tra generazioni diverse. Quindi è una buona idea salvare e ripristinare sequence_lengths: un vettore di lunghezze di ogni sequenza di input. output: Se time_major è vero, questo è un tensore 3-D con la forma di [seq_length, batch_size, dir * num_units]. Se time_major è false, la forma è [batch_size, seq_length, dir * num_units]. output_h: la stessa forma ha input_h. output_c: la stessa forma di input_c per LSTM. Un tensore vuoto per altri modelli. output_backprop: un tensore 3-D con la stessa forma dell'output nel passaggio in avanti. output_h_backprop: un tensore 3-D con la stessa forma di output_h nel passaggio in avanti. output_c_backprop: un tensore 3-D con la stessa forma di output_c nel passaggio in avanti. time_major: indica se il formato di input / output è time major o batch major. reserve_space: lo stesso reserve_space prodotto nell'operazione di inoltro. input_backprop: il backprop da inserire nel passaggio in avanti. Ha la stessa forma dell'input. input_h_backprop: il backprop di input_h nel passaggio in avanti. Ha la stessa forma di input_h. input_c_backprop: il backprop di input_c nel passaggio in avanti. Ha la stessa forma di input_c. params_backprop: il backprop nel buffer dei parametri nel passaggio in avanti. Ha la stessa forma dei parametri.

Classi annidate

classe CudnnRNNBackpropV3.Options Attributi facoltativi per CudnnRNNBackpropV3

Metodi pubblici

static <T extends Number> CudnnRNNBackpropV3 <T>
create ( Scope scope, Operando <T> input, Operand <T> inputH, Operand <T> inputC, Operand <T> params, Operand <Integer> sequenceLengths, Operand <T> output, Operand <T> outputH, Operand <T > outputC, Operando <T> outputBackprop, Operando <T> outputHBackprop, Operando <T> outputCBackprop, Operando <T> reserveSpace, Operando <?> hostReserved, Opzioni ... opzioni)
Metodo Factory per creare una classe che racchiude una nuova operazione CudnnRNNBackpropV3.
static CudnnRNNBackpropV3.Options
static CudnnRNNBackpropV3.Options
dropout (Float dropout)
Uscita <T>
Uscita <T>
Uscita <T>
static CudnnRNNBackpropV3.Options
inputMode (stringa inputMode)
static CudnnRNNBackpropV3.Options
numProj (lungo numProj)
Uscita <T>
static CudnnRNNBackpropV3.Options
rnnMode (stringa rnnMode)
static CudnnRNNBackpropV3.Options
seme ( seme lungo)
static CudnnRNNBackpropV3.Options
seed2 ( seme lungo2)
static CudnnRNNBackpropV3.Options
timeMajor (Boolean timeMajor)

Metodi ereditati

Metodi pubblici

public static CudnnRNNBackpropV3 <T> create ( Scope scope, Operand <T> input, Operand <T> inputH, Operand <T> inputC, Operand <T> params, Operand <Integer> sequenceLengths, Operand <T> output, Operand <T > outputH, Operando <T> outputC, Operando <T> outputBackprop, Operando <T> outputHBackprop, Operando <T> outputCBackprop, Operando <T> reserveSpace, Operando <?> hostReserved, Opzioni ... opzioni)

Metodo Factory per creare una classe che racchiude una nuova operazione CudnnRNNBackpropV3.

Parametri
scopo ambito attuale
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di CudnnRNNBackpropV3

public static CudnnRNNBackpropV3.Options direction (String direction)

public static CudnnRNNBackpropV3.Options dropout (Float dropout)

output pubblico <T> inputBackprop ()

output pubblico <T> inputCBackprop ()

output pubblico <T> inputHBackprop ()

public static CudnnRNNBackpropV3.Options inputMode (String inputMode)

public static CudnnRNNBackpropV3.Options numProj (Long numProj)

output pubblico <T> paramsBackprop ()

public static CudnnRNNBackpropV3.Options rnnMode (String rnnMode)

public static CudnnRNNBackpropV3.Options seed (seme lungo)

public static CudnnRNNBackpropV3.Options seed2 (Long seed2)

public static CudnnRNNBackpropV3.Options timeMajor (Boolean timeMajor)