CudnnRNNParamsToCanonicalV2

classe finale pubblica CudnnRNNParamsToCanonicalV2

Recupera i parametri CudnnRNN in forma canonica. Supporta la proiezione in LSTM.

Recupera un set di pesi dal buffer params opaco che può essere salvato e ripristinato in modo compatibile con le esecuzioni future.

Tieni presente che il buffer dei parametri potrebbe non essere compatibile tra GPU diverse. Quindi qualsiasi salvataggio e restauro dovrebbe essere convertito da e verso i pesi e i pregiudizi canonici.

num_layers: specifica il numero di livelli 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 gli strati. num_params_biases: numero di vettori dei parametri di bias per tutti i livelli. pesi: la forma canonica dei pesi che possono essere utilizzati per il salvataggio e il restauro. È più probabile che siano compatibili tra generazioni diverse. bias: la forma canonica dei bias che può essere utilizzata per il salvataggio e il restauro. È più probabile che siano compatibili tra generazioni diverse. rnn_mode: indica il tipo del 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; altrimenti implica 'linear_input'. direzione: indica se verrà utilizzato un modello bidirezionale. dir = (direzione == bidirezionale) ? 2:1 dropout: probabilità di dropout. Quando impostato su 0., il dropout è disabilitato. seme: la prima parte di un seme per inizializzare il dropout. seed2: la seconda parte di un seed per inizializzare il dropout. num_proj: la dimensionalità dell'output per le matrici di proiezione. Se Nessuno o 0, non viene eseguita alcuna proiezione.

Classi nidificate

classe CudnnRNNParamsToCanonicalV2.Options Attributi facoltativi per CudnnRNNParamsToCanonicalV2

Metodi pubblici

Elenco< Uscita <T>>
static <T estende il numero> CudnnRNNParamsToCanonicalV2 <T>
create ( ambito ambito , operando <intero> numLayers, operando <intero> numUnits, operando <intero> inputSize, operando <T> parametri, lungo numParamsWeights, lungo numParamsBiases, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CudnnRNNParamsToCanonicalV2.
statico CudnnRNNParamsToCanonicalV2.Options
direzione (direzione della stringa)
statico CudnnRNNParamsToCanonicalV2.Options
dropout (dropout flottante)
statico CudnnRNNParamsToCanonicalV2.Options
inputMode (Stringa inputMode)
statico CudnnRNNParamsToCanonicalV2.Options
numProj (numProj lungo)
statico CudnnRNNParamsToCanonicalV2.Options
rnnMode (Stringa rnnMode)
statico CudnnRNNParamsToCanonicalV2.Options
seme (seme lungo)
statico CudnnRNNParamsToCanonicalV2.Options
seme2 (seme lungo2)
Elenco< Uscita <T>>
pesi ()

Metodi ereditati

Metodi pubblici

Elenco pubblico< Output <T>> bias ()

public static CudnnRNNParamsToCanonicalV2 <T> create ( Scope scope, Operando <Integer> numLayers, Operando <Integer> numUnits, Operando <Integer> inputSize, Operando <T> params, Long numParamsWeights, Long numParamsBiases, Opzioni... opzioni)

Metodo factory 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

direzione pubblica statica CudnnRNNParamsToCanonicalV2.Options (direzione stringa)

abbandono pubblico statico CudnnRNNParamsToCanonicalV2.Options (abbandono mobile)

public static CudnnRNNParamsToCanonicalV2.Options inputMode (String inputMode)

pubblico statico CudnnRNNParamsToCanonicalV2.Options numProj (numProj lungo)

pubblico statico CudnnRNNParamsToCanonicalV2.Options rnnMode (String rnnMode)

seme statico pubblico CudnnRNNParamsToCanonicalV2.Options (seme lungo)

pubblico statico CudnnRNNParamsToCanonicalV2.Options seed2 (seme lungo2)

public List< Output <T>> pesi ()