Converte i parametri CudnnRNN dalla forma canonica alla forma utilizzabile. Supporta la proiezione in LSTM.
Scrive un set di pesi nel buffer opaco dei parametri in modo che possano essere utilizzati nell'addestramento o nelle inferenze successive.
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. 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. 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. 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 | CudnnRNNCanonalToParams.Options | Attributi facoltativi per CudnnRNNCanonicalToParams |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita <T> | comeuscita () Restituisce l'handle simbolico del tensore. |
static <T estende TNumber > CudnnRNNCanonalToParams <T> | |
statico CudnnRNNCanonalToParams.Options | direzione (direzione della stringa) |
statico CudnnRNNCanonalToParams.Options | dropout (dropout flottante) |
statico CudnnRNNCanonalToParams.Options | inputMode (Stringa inputMode) |
statico CudnnRNNCanonalToParams.Options | numProj (numProj lungo) |
Uscita <T> | parametri () |
statico CudnnRNNCanonalToParams.Options | rnnMode (Stringa rnnMode) |
statico CudnnRNNCanonalToParams.Options | seme (seme lungo) |
statico CudnnRNNCanonalToParams.Options | seme2 (seme lungo2) |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce l'handle simbolico del tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static CudnnRNNCanonicalToParams <T> create ( ambito scope, operando < TInt32 > numLayers, operando < TInt32 > numUnits, operando < TInt32 > inputSize, pesi Iterable< operando <T>>, bias Iterable< operando <T>>, opzioni.. . opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CudnnRNNCanonalToParams.
Parametri
scopo | ambito attuale |
---|---|
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CudnnRNNCanonalToParams