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 | CudnnRNNCanonalToParamsV2.Options | Attributi facoltativi per CudnnRNNCanonicalToParamsV2 | |
Metodi pubblici
| Uscita <T> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
| static <T estende il numero> CudnnRNNCanonalToParamsV2 <T> | |
| statico CudnnRNNCanonalToParamsV2.Options | direzione (direzione della stringa) |
| statico CudnnRNNCanonalToParamsV2.Options | dropout (dropout flottante) |
| statico CudnnRNNCanonalToParamsV2.Options | inputMode (Stringa inputMode) |
| statico CudnnRNNCanonalToParamsV2.Options | numProj (numProj lungo) |
| Uscita <T> | parametri () |
| statico CudnnRNNCanonalToParamsV2.Options | rnnMode (Stringa rnnMode) |
| statico CudnnRNNCanonalToParamsV2.Options | seme (seme lungo) |
| statico CudnnRNNCanonalToParamsV2.Options | seme2 (seme lungo2) |
Metodi ereditati
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce la maniglia simbolica di un 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 CudnnRNNCanonicalToParamsV2 <T> create ( ambito ambito , operando <intero> numLayers, operando <intero> numUnits, operando <intero> inputSize, pesi Iterable< operando <T>>, bias Iterable< operando <T>>, opzioni.. . opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CudnnRNNCanonalToParamsV2.
Parametri
| scopo | ambito attuale |
|---|---|
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CudnnRNNCanonalToParamsV2