Calcola la dimensione dei pesi che possono essere utilizzati da un modello Cudnn RNN.
Restituisce la dimensione dei parametri che può essere utilizzata dal modello Cudnn RNN. L'allocazione e l'inizializzazione successive del peso dovrebbero utilizzare questa dimensione.
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. 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. params_size: la dimensione del buffer dei parametri che deve essere allocato e inizializzato per questo modello RNN. Tieni presente che questo buffer dei parametri potrebbe non essere compatibile tra le GPU. Utilizza CudnnRNNParamsWeights e CudnnRNNParamsBiases per salvarli e ripristinarli in modo compatibile tra le diverse esecuzioni.
Classi nidificate
classe | CudnnRnnParamsSize.Options | Attributi facoltativi per CudnnRnnParamsSize |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita <U> | comeuscita () Restituisce l'handle simbolico del tensore. |
statico <U estende TNumber , T estende TNumber > CudnnRnnParamsSize <U> | |
statico CudnnRnnParamsSize.Options | direzione (direzione della stringa) |
statico CudnnRnnParamsSize.Options | dropout (dropout flottante) |
statico CudnnRnnParamsSize.Options | inputMode (Stringa inputMode) |
statico CudnnRnnParamsSize.Options | numProj (numProj lungo) |
Uscita <U> | |
statico CudnnRnnParamsSize.Options | rnnMode (Stringa rnnMode) |
statico CudnnRnnParamsSize.Options | seme (seme lungo) |
statico CudnnRnnParamsSize.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 <U> 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 CudnnRnnParamsSize <U> create ( Scope scope, Operando < TInt32 > numLayers, Operando < TInt32 > numUnits, Operando < TInt32 > inputSize, Class<T> T, Class<U> S, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CudnnRnnParamsSize.
Parametri
scopo | ambito attuale |
---|---|
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CudnnRnnParamsSize