Recupera i parametri CudnnRNN in forma canonica. Supporta la proiezione in LSTM.
Recupera un set di pesi dal buffer dei parametri opaco che può essere salvato e ripristinato in un modo compatibile con le esecuzioni future.
Tieni presente che il buffer params potrebbe non essere compatibile con GPU diverse. Quindi qualsiasi salvataggio e ripristino dovrebbe essere convertito da e verso i pesi e i pregiudizi canonici.
num_layers: specifica il numero di strati nel modello RNN. num_units: specifica la dimensione dello stato nascosto. input_size: specifica la dimensione dello stato di input. num_params_weights: numero della matrice dei parametri di peso per tutti i livelli. num_params_biases: numero di vettore parametro bias per tutti i livelli. pesi: la forma canonica dei pesi che possono essere utilizzati per il salvataggio e il ripristino. È più probabile che siano compatibili tra generazioni diverse. bias: la forma canonica di bias che può essere utilizzata per il salvataggio e il ripristino. È più probabile che siano compatibili tra generazioni diverse. rnn_mode: indica il tipo di modello RNN. input_mode: indica se c'è una proiezione lineare tra l'input e il calcolo effettivo prima del primo strato. '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. dir = (direction == bidirezionale)? Abbandono 2: 1: probabilità di abbandono. Quando è impostato su 0., il dropout è disabilitato. seme: la prima parte di un seme per inizializzare l'abbandono. seed2: la seconda parte di un seme per inizializzare l'abbandono. num_proj: la dimensionalità di output per le matrici di proiezione. Se Nessuno o 0, non viene eseguita alcuna proiezione.
Classi annidate
classe | CudnnRNNParamsToCanonicalV2.Options | Attributi facoltativi per CudnnRNNParamsToCanonicalV2 |
Metodi pubblici
Elenco < Uscita <T>> | pregiudizi () |
static <T extends Number> CudnnRNNParamsToCanonicalV2 <T> | |
static CudnnRNNParamsToCanonicalV2.Options | direction ( direzione della stringa) |
static CudnnRNNParamsToCanonicalV2.Options | dropout (Float dropout) |
static CudnnRNNParamsToCanonicalV2.Options | inputMode (stringa inputMode) |
static CudnnRNNParamsToCanonicalV2.Options | numProj (lungo numProj) |
static CudnnRNNParamsToCanonicalV2.Options | rnnMode (stringa rnnMode) |
static CudnnRNNParamsToCanonicalV2.Options | |
static CudnnRNNParamsToCanonicalV2.Options | |
Elenco < Uscita <T>> | pesi () |
Metodi ereditati
Metodi pubblici
public static CudnnRNNParamsToCanonicalV2 <T> create ( Scope scope, Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options ... options)
Metodo di fabbrica per creare una classe che racchiude una nuova operazione CudnnRNNParamsToCanonicalV2.
Parametri
scopo | ambito attuale |
---|---|
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CudnnRNNParamsToCanonicalV2