Comment utiliser l’opération d’expansion dans PyTorch ?

PyTorch est une bibliothèque d'apprentissage automatique qui permet aux utilisateurs de travailler avec des tenseurs. Les tenseurs sont des structures de données essentielles qui peuvent avoir différents types de données et de formes. Parfois, les utilisateurs peuvent souhaiter effectuer l’opération d’expansion sur les tenseurs pour étendre leur taille. L'opération d'expansion permet aux utilisateurs de répéter un tenseur selon des dimensions spécifiées. PyTorch fournit l'attribut « expand() » qui prend un tenseur et une liste de tailles comme entrées. Il renvoie un nouveau tenseur qui a les mêmes données mais avec des dimensions différentes.

Cet article illustrera la méthode pour utiliser l'opération d'expansion sur les tenseurs dans PyTorch.

Comment utiliser l’opération d’expansion dans PyTorch ?

Pour utiliser l'opération d'expansion dans PyTorch, consultez les étapes ci-dessous :

• Importer la bibliothèque PyTorch
• Créer un tenseur souhaité
• Afficher la taille du tenseur d'entrée
• Développez le tenseur en utilisant le ' développer() ' attribut
• Afficher le tenseur étendu et sa taille

Étape 1 : Importer la bibliothèque PyTorch

Tout d’abord, importez le « torche ' bibliothèque pour utiliser l'opération d'expansion :

Étape 2 : créer un tenseur

Ensuite, créez le tenseur souhaité en utilisant le ' torche.tensor() »fonctionner et imprimer ses éléments. Ici, nous créons ce qui suit » Dizaines ' Tenseur :

Dans le résultat ci-dessous, le tenseur a été créé avec succès :

Étape 3 : Afficher la taille du tenseur d'entrée

Ensuite, affichez la taille du « créé ci-dessus » Dizaines ' Tenseur utilisant le ' taille() ' attribut:

D'après le résultat ci-dessous, la taille du tenseur est de 3 × 1 :

Étape 4 : développer le tenseur

Maintenant, utilisez le ' développer() ' pour effectuer l'opération d'expansion et étendre le tenseur à une nouvelle dimension. Il est nécessaire de fournir le tenseur et la liste des tailles en entrée. Ici, nous étendons le tenseur à la taille 3×4 :

Étape 5 : Afficher le tenseur étendu et sa taille

Enfin, imprimez les éléments du tenseur développé et sa taille :

La sortie ci-dessous montre le tenseur étendu et sa taille, c'est-à-dire 3 × 4. Cela indique que l'opération étendue a été effectuée avec succès :

Nous avons expliqué la méthode efficace pour utiliser l'opération d'expansion dans PyTorch

Note  : Vous pouvez accéder à notre bloc-notes Google Colab à cette adresse lien .

Conclusion

Pour utiliser l'opération d'expansion dans PyTorch, importez d'abord la bibliothèque Torch. Ensuite, créez le tenseur souhaité et affichez ses éléments et sa taille. Ensuite, utilisez le ' développer() ' Attribut pour développer le tenseur d'entrée. Enfin, imprimez le tenseur développé et affichez sa taille. Cet article a démontré la méthode pour utiliser l'opération d'expansion sur les tenseurs dans PyTorch.