Le TF.Fonction de pile dans TensorFlow.Js

Dans le Tensorflow.bibliothèque JS, le TF.La fonction stack () est utilisée pour rejoindre deux ou plusieurs tenseurs.

Syntaxe:

Paramètres:

1. Il faut deux tenseurs ou plus comme paramètre qui peut être unidimensionnel comme premier paramètre.
2. L'axe prend deux valeurs possibles (0 et 1).

Si spécifié 0, TF.stack () rejoint les tenseurs les uns après les autres en renvoyant un nouveau tenseur.

Si spécifié 1, TF.stack () rejoint l'élément par élément d'affilée en renvoyant un nouveau tenseur.

Exemple 1

Créez deux tenseurs 1D avec des entiers et empilez deux tenseurs en définissant l'axe-0.

Sortir:

Tensor1 et Tensor2 sont empilés l'un après l'autre.

Exemple 2

Créez deux tenseurs 1D avec des entiers et empilez deux tenseurs en définissant l'axe-1.

Sortir:

Tensor1 et Tensor2 sont un élément empilé par élément dans les deux tenseurs, et le tenseur renvoyé a des éléments empilés placés dans des lignes séparées.

Exemple 3

Créez quatre tenseurs 1D avec des entiers et empilez-les en définissant AXIS-1 et AXIS-0 séparément.

Sortir:

Dans la première sortie, quatre tenseurs sont empilés par élément, et dans la deuxième sortie, les tenseurs sont empilés à la fois.

Conclusion

Dans ce Tensorflow.Tutoriel JS, nous avons vu comment empiler deux tenseurs ou plus en utilisant le TF.Fonction Stack (). Si l'axe est spécifié comme 0, TF.stack () rejoint les tenseurs les uns après les autres en renvoyant un nouveau tenseur. Si spécifié comme 1, TF.stack () rejoint l'élément par élément d'affilée en renvoyant un nouveau tenseur. Assurez-vous de comprendre tous les exemples discutés et d'obtenir la différence entre les paramètres AXIS-0/1.