CudnnRNNCanonicalToParamsV2

public final class CudnnRNNCanonicalToParamsV2

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
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
seme ( seme lungo)
static CudnnRNNCanonicalToParamsV2.Options
seed2 ( seme lungo2)

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

public static CudnnRNNCanonicalToParamsV2.Options direction (String direction)

public static CudnnRNNCanonicalToParamsV2.Options dropout (Float dropout)

public static CudnnRNNCanonicalToParamsV2.Options inputMode (String inputMode)

public static CudnnRNNCanonicalToParamsV2.Options numProj (Long numProj)

public Output <T> params ()

public static CudnnRNNCanonicalToParamsV2.Options rnnMode (String rnnMode)

public static CudnnRNNCanonicalToParamsV2.Options seed (seme lungo)

public static CudnnRNNCanonicalToParamsV2.Options seed2 (Long seed2)