Converte i parametri CudnnRNN dalla forma canonica alla forma utilizzabile. Supporta la proiezione in LSTM.
Scrive una serie di pesi nel buffer dei parametri opaco in modo che possano essere utilizzati nell'addestramento o nelle inferenze imminenti.
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. 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. 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. 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. Se 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 | CudnnRNNCanonicalToParamsV2.Options | Attributi facoltativi per CudnnRNNCanonicalToParamsV2 |
Metodi pubblici
Uscita <T> | asOutput () Restituisce la maniglia simbolica di un tensore. |
static <T extends Number> CudnnRNNCanonicalToParamsV2 <T> | create ( ambito ambito, operando <Integer> numLayers, operando <Integer> numUnits, operando <Integer> inputSize, Iterable < Operand <T>> weights, Iterable < Operand <T>> biases, Options ... options) Metodo di fabbrica per creare una classe che racchiude una nuova operazione CudnnRNNCanonicalToParamsV2. |
static CudnnRNNCanonicalToParamsV2.Options | direction ( direzione della stringa) |
static CudnnRNNCanonicalToParamsV2.Options | dropout (Float dropout) |
static CudnnRNNCanonicalToParamsV2.Options | inputMode (stringa inputMode) |
static CudnnRNNCanonicalToParamsV2.Options | numProj (lungo numProj) |
Uscita <T> | params () |
static CudnnRNNCanonicalToParamsV2.Options | rnnMode (stringa rnnMode) |
static CudnnRNNCanonicalToParamsV2.Options | |
static CudnnRNNCanonicalToParamsV2.Options |
Metodi ereditati
Metodi pubblici
output pubblico <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 ( Scope scope, Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Iterable < Operand <T>> weights, Iterable < Operand <T>> biases, Options .. . opzioni)
Metodo di fabbrica per creare una classe che racchiude una nuova operazione CudnnRNNCanonicalToParamsV2.
Parametri
scopo | ambito attuale |
---|---|
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CudnnRNNCanonicalToParamsV2