CudnnRNNParamsToCanonicalV2

public final class CudnnRNNParamsToCanonicalV2

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

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

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. 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. 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. 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. Quando è 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 CudnnRNNParamsToCanonicalV2.Options Attributi facoltativi per CudnnRNNParamsToCanonicalV2

Metodi pubblici

Elenco < Uscita <T>>
static <T extends Number> CudnnRNNParamsToCanonicalV2 <T>
create ( Scope scope, Operando <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Opzioni ... opzioni)
Metodo di fabbrica per creare una classe che racchiude una nuova operazione CudnnRNNParamsToCanonicalV2.
static CudnnRNNParamsToCanonicalV2.Options
static CudnnRNNParamsToCanonicalV2.Options
dropout (Float dropout)
static CudnnRNNParamsToCanonicalV2.Options
inputMode (stringa inputMode)
static CudnnRNNParamsToCanonicalV2.Options
numProj (lungo numProj)
static CudnnRNNParamsToCanonicalV2.Options
rnnMode (stringa rnnMode)
static CudnnRNNParamsToCanonicalV2.Options
seme ( seme lungo)
static CudnnRNNParamsToCanonicalV2.Options
seed2 ( seme lungo2)
Elenco < Uscita <T>>
pesi ()

Metodi ereditati

Metodi pubblici

public List < Output <T>> bias ()

public static CudnnRNNParamsToCanonicalV2 <T> create ( Scope scope, Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options ... options)

Metodo di fabbrica 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

public static CudnnRNNParamsToCanonicalV2.Options direction (String direction)

public static CudnnRNNParamsToCanonicalV2.Options dropout (Float dropout)

public static CudnnRNNParamsToCanonicalV2.Options inputMode (String inputMode)

public static CudnnRNNParamsToCanonicalV2.Options numProj (Long numProj)

public static CudnnRNNParamsToCanonicalV2.Options rnnMode (String rnnMode)

public static CudnnRNNParamsToCanonicalV2.Options seed (seme lungo)

public static CudnnRNNParamsToCanonicalV2.Options seed2 (Long seed2)

public List < Output <T>> weights ()