CudnnRNNCanonicalToParamsV2

classe finale publique CudnnRNNCanonicalToParamsV2

Convertit les paramètres CudnnRNN de la forme canonique en forme utilisable. Il prend en charge la projection dans LSTM.

Écrit un ensemble de poids dans le tampon de paramètres opaque afin qu'ils puissent être utilisés dans les formations ou inférences à venir.

Notez que le tampon de paramètres peut ne pas être compatible entre différents GPU. Ainsi, toute sauvegarde et restauration doit être convertie vers et depuis les poids et biais canoniques.

num_layers : spécifie le nombre de couches dans le modèle RNN. num_units : spécifie la taille de l'état caché. input_size : Spécifie la taille de l’état d’entrée. poids : la forme canonique des poids qui peuvent être utilisés pour la sauvegarde et la restauration. Ils sont plus susceptibles d’être compatibles entre les différentes générations. biais : la forme canonique des biais qui peuvent être utilisés pour la sauvegarde et la restauration. Ils sont plus susceptibles d’être compatibles entre les différentes générations. num_params_weights : nombre de matrices de paramètres de poids pour toutes les couches. num_params_biases : nombre de vecteurs de paramètres de biais pour toutes les couches. rnn_mode : Indique le type du modèle RNN. input_mode : indique s'il existe une projection linéaire entre l'entrée et le calcul réel avant la première couche. 'skip_input' n'est autorisé que lorsque input_size == num_units ; 'auto_select' implique 'skip_input' lorsque input_size == num_units ; sinon, cela implique 'linear_input'. direction : Indique si un modèle bidirectionnel sera utilisé. dir = (direction == bidirectionnel) ? 2 : 1 abandon : probabilité d’abandon. Lorsqu'il est défini sur 0, la suppression est désactivée. seed : la 1ère partie d'une graine pour initialiser le dropout. seed2 : la 2ème partie d'une graine pour initialiser l'abandon. num_proj : la dimensionnalité de sortie pour les matrices de projection. Si Aucun ou 0, aucune projection n’est effectuée.

Classes imbriquées

classe CudnnRNNCanonicalToParamsV2.Options Attributs facultatifs pour CudnnRNNCanonicalToParamsV2

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T étend le nombre> CudnnRNNCanonicalToParamsV2 <T>
créer ( Portée de portée , Opérande <Integer> numLayers, Opérande <Integer> numUnits, Opérande <Integer> inputSize, Iterable< Operand <T>> poids, Iterable< Operand <T>> biais, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CudnnRNNCanonicalToParamsV2.
statique CudnnRNNCanonicalToParamsV2.Options
direction (direction de la chaîne)
statique CudnnRNNCanonicalToParamsV2.Options
abandon (abandon flottant)
statique CudnnRNNCanonicalToParamsV2.Options
inputMode (chaîne inputMode)
statique CudnnRNNCanonicalToParamsV2.Options
numProj (numProj long)
Sortie <T>
statique CudnnRNNCanonicalToParamsV2.Options
rnnMode (Chaîne rnnMode)
statique CudnnRNNCanonicalToParamsV2.Options
graine (graine longue)
statique CudnnRNNCanonicalToParamsV2.Options
graine2 (graine longue2)

Méthodes héritées

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique d'un tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static CudnnRNNCanonicalToParamsV2 <T> créer ( Portée de portée , Opérande <Integer> numLayers, Opérande <Integer> numUnits, Opérande <Integer> inputSize, Iterable< Operand <T>> poids, Iterable< Operand <T>> biais, Options. .options )

Méthode d'usine pour créer une classe encapsulant une nouvelle opération CudnnRNNCanonicalToParamsV2.

Paramètres
portée portée actuelle
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de CudnnRNNCanonicalToParamsV2

direction statique publique CudnnRNNCanonicalToParamsV2.Options (direction de la chaîne)

abandon public statique CudnnRNNCanonicalToParamsV2.Options (abandon flottant)

public statique CudnnRNNCanonicalToParamsV2.Options inputMode (String inputMode)

public statique CudnnRNNCanonicalToParamsV2.Options numProj (Long numProj)

paramètres de sortie publics <T> ()

public statique CudnnRNNCanonicalToParamsV2.Options rnnMode (String rnnMode)

graine publique statique CudnnRNNCanonicalToParamsV2.Options (graine longue)

public statique CudnnRNNCanonicalToParamsV2.Options seed2 (Long seed2)