Un RNN soutenu par cuDNN.
Calcule le RNN à partir des états d'entrée et initiaux, par rapport au tampon de paramètres. Accepte une entrée supplémentaire "sequence_lengths" que CudnnRNN.
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é. Doit être « unidirectionnel » ou « bidirectionnel ». 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 l'abandon. seed2 : la 2ème partie d'une graine pour initialiser l'abandon. input : Si time_major est vrai, il s'agit d'un tenseur 3D avec la forme de [seq_length, batch_size, input_size]. Si time_major est faux, la forme est [batch_size, seq_length, input_size]. input_h : Si time_major est vrai, il s'agit d'un tenseur 3D avec la forme de [num_layer * dir, batch_size, num_units]. Si time_major est faux, la forme est [batch_size, num_layer * dir, num_units]. input_c : pour LSTM, un tenseur 3D avec la forme de [num_layer * dir, batch, num_units]. Pour les autres modèles, il est ignoré. params : un tenseur 1D qui contient les poids et les biais dans une présentation opaque. La taille doit être créée via CudnnRNNParamsSize et initialisée séparément. Notez qu’ils peuvent ne pas être compatibles entre différentes générations. C'est donc une bonne idée de sauvegarder et de restaurer séquence_lengths : un vecteur de longueurs de chaque séquence d'entrée. sortie : si time_major est vrai, il s'agit d'un tenseur 3D ayant la forme de [seq_length, batch_size, dir * num_units]. Si time_major est faux, la forme est [batch_size, seq_length, dir * num_units]. output_h : la même forme a input_h. output_c : la même forme que input_c pour LSTM. Un tenseur vide pour d'autres modèles. is_training : indique si cette opération est utilisée pour l'inférence ou la formation. time_major : indique si le format d'entrée/sortie est majeur en temps ou majeur en lot. reserve_space : Un tenseur opaque qui peut être utilisé dans le calcul du backprop. Il n'est produit que si is_training est vrai.
Classes imbriquées
| classe | CudnnRNN.Options | Attributs facultatifs pour CudnnRNN | |
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| statique <T étend TNumber > CudnnRNN <T> | créer ( Portée de portée , Entrée opérande <T>, Opérande <T> inputH, Opérande <T> inputC, Paramètres d'opérande <T>, Opérande < TInt32 > séquenceLengths, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération CudnnRNN. |
| CudnnRNN.Options statiques | direction (direction de la chaîne) |
| CudnnRNN.Options statiques | abandon (abandon flottant) |
| Sortie <?> | hôteRéservé () |
| CudnnRNN.Options statiques | inputMode (chaîne inputMode) |
| CudnnRNN.Options statiques | isTraining (isTraining booléen) |
| CudnnRNN.Options statiques | numProj (numProj long) |
| Sortie <T> | sortir () |
| Sortie <T> | sortieC () |
| Sortie <T> | sortieH () |
| Sortie <T> | |
| CudnnRNN.Options statiques | rnnMode (Chaîne rnnMode) |
| CudnnRNN.Options statiques | graine (graine longue) |
| CudnnRNN.Options statiques | graine2 (graine longue2) |
| CudnnRNN.Options statiques | timeMajor (Booléen timeMajor) |
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 CudnnRNN <T> créer ( Portée de portée , Opérande <T> entrée, Opérande <T> inputH, Opérande <T> inputC, Paramètres d' opérande <T>, Opérande < TInt32 > séquenceLengths, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CudnnRNN.
Paramètres
| portée | portée actuelle |
|---|---|
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de CudnnRNN