Récupère les paramètres CudnnRNN sous forme canonique. Il prend en charge la projection dans LSTM.
Récupère un ensemble de poids du tampon de paramètres opaque qui peut être enregistré et restauré d'une manière compatible avec les exécutions futures.
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. 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. 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. 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 | CudnnRNNParamsToCanonical.Options | Attributs facultatifs pour CudnnRNNParamsToCanonical | |
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| Liste< Sortie <T>> | Les préjugés () |
| statique <T étend TNumber > CudnnRNNParamsToCanonical <T> | |
| statique CudnnRNNParamsToCanonical.Options | direction (direction de la chaîne) |
| statique CudnnRNNParamsToCanonical.Options | abandon (abandon flottant) |
| statique CudnnRNNParamsToCanonical.Options | inputMode (chaîne inputMode) |
| statique CudnnRNNParamsToCanonical.Options | numProj (numProj long) |
| statique CudnnRNNParamsToCanonical.Options | rnnMode (Chaîne rnnMode) |
| statique CudnnRNNParamsToCanonical.Options | graine (graine longue) |
| statique CudnnRNNParamsToCanonical.Options | graine2 (graine longue2) |
| Liste< Sortie <T>> | poids () |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
public static CudnnRNNParamsToCanonical <T> créer ( Scope scope, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CudnnRNNParamsToCanonical.
Paramètres
| portée | portée actuelle |
|---|---|
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de CudnnRNNParamsToCanonical