Pytorch - carré

Nous retournerons le carré de tous les éléments de Tensor en utilisant la méthode carrée () dans ce tutoriel pytorch.

Pytorch est un cadre open source disponible avec un langage de programmation Python. Tensor est un tableau multidimensionnel utilisé pour stocker les données. Pour utiliser un tenseur, nous devons importer le module de torche. Pour créer un tenseur, la méthode utilisée est tenseur ().

Syntaxe:

torche.tenseur (données)

Où les données sont un tableau multidimensionnel.

Torche.carré()

Carré () en pytorch renvoie le carré de chaque élément dans le tenseur de pytorch. Il faut un paramètre.

Syntaxe:

torche.Square (Tensor_Object)

Paramètre:

Tensor_Object est un tenseur

Exemple 1:

Dans cet exemple, nous créerons un tenseur avec une dimension qui a 5 éléments et retournerons le carré de ces 5 éléments dans un tenseur.

Module de torche #mport
Importer une torche
#create un tenseur
data1 = torche.Tensor ([12,34,56,1,10])
#afficher
Imprimer ("Tensor réel:")
Imprimer (DATA1)
imprimer ("carré:")
imprimer (torche.carré (data1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Carré:
Tensor ([144, 1156, 3136, 1, 100])

Fonctionnement:

1. 12 * 12 = 144
2. 34 * 34 = 1156
3. 56 * 56 = 3136
4. 1 * 1 = 1
5. 10 * 10 = 100

Exemple 2:

Dans cet exemple, nous créerons un tenseur avec deux dimensions qui a 5 éléments dans chaque rangée et retournerons les carrés des éléments.

Module de torche #mport
Importer une torche
#create un tenseur 2D
data1 = torche.Tensor ([[45,67,21,23,2], [- 2,3,4,5,6]])
#afficher
Imprimer ("Tensor réel:")
Imprimer (DATA1)
imprimer ("carré")
imprimer (torche.carré (data1))

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[-2, 3, 4, 5, 6]])
Carré
Tensor ([[2025, 4489, 441, 529, 4],
[4, 9, 16, 25, 36]])

Fonctionnement:

1. 45 * 45 = 2025, -2 * -2 = 4
2. 67 * 67 = 4489,3 * 3 = 9
3. 21 * 21 = 441,4 * 4 = 16
4. 23 * 23 = 529,5 * 5 = 25
5. 2 * 2 = 4,6 * 6 = 36

Travailler avec le processeur

Si vous souhaitez exécuter une fonction carrée () sur le CPU, nous devons créer un tenseur avec une fonction CPU (). Cela fonctionnera sur une machine CPU.

Lorsque nous créons un tenseur, cette fois, nous pouvons utiliser la fonction CPU ().

Syntaxe:

torche.tenseur (données).CPU()

Exemple 1:

Dans cet exemple, nous créerons un tenseur avec une dimension qui a 5 éléments sur la fonction CPU et retournerons le carré de ces 5 éléments dans un tenseur.

Module de torche #mport
Importer une torche
#create un tenseur
data1 = torche.Tensor ([12,34,56,1,10]).CPU()
#afficher
Imprimer ("Tensor réel:")
Imprimer (DATA1)
imprimer ("carré:")
imprimer (torche.carré (data1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Carré:
Tensor ([144, 1156, 3136, 1, 100])

Fonctionnement:

1. 12 * 12 = 144
2. 34 * 34 = 1156
3. 56 * 56 = 3136
4. 1 * 1 = 1
5. 10 * 10 = 100

Exemple 2:

Dans cet exemple, nous créerons un tenseur avec deux dimensions qui a 5 éléments sur une fonction CPU dans chaque ligne et retourner les carrés des éléments.

Module de torche #mport
Importer une torche
#create un tenseur 2D
data1 = torche.Tensor ([[45,67,21,23,2], [- 2,3,4,5,6]]).CPU()
#afficher
Imprimer ("Tensor réel:")
Imprimer (DATA1)
imprimer ("carré")
imprimer (torche.carré (data1))

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[-2, 3, 4, 5, 6]])
Carré
Tensor ([[2025, 4489, 441, 529, 4],
[4, 9, 16, 25, 36]])

Fonctionnement:

1. 45 * 45 = 2025, -2 * -2 = 4
2. 67 * 67 = 4489,3 * 3 = 9
3. 21 * 21 = 441,4 * 4 = 16
4. 23 * 23 = 529,5 * 5 = 25
5. 2 * 2 = 4,6 * 6 = 36

Conclusion

Dans cette leçon de pytorch, nous avons discuté de la fonction carrée (). Il renvoie le carré de chaque élément du tenseur pytorche. Nous avons discuté des deux exemples avec les différents tenseurs dimensionnels pour effectuer la fonction carrée ().