Pytorch - sqrt

Nous retournerons la racine carrée de tous les éléments du tenseur en utilisant la méthode SQRT () 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.SQRT ()

Sqrt () en pytorch renvoie la racine carrée de chaque élément du tenseur pytorch. Il faut un paramètre.

Syntaxe:

torche.SQRT (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 les racines carrées 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 ("Root carré:")
imprimer (torche.SQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Racine carrée:
tenseur ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Fonctionnement:

1. √12 = 3.4641
2. √34 = 5.8310
3. √56 = 7.4833
4. √1 = 1.0000
5. √10 = 3.1623

Exemple 2:

Dans cet exemple, nous créerons un tenseur avec deux dimensions qui a 5 éléments dans chaque ligne et renvoyer la racine carrée 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 ("Root carré:")
imprimer (torche.SQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Racine carrée:
tenseur ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Fonctionnement:

1. √45 = 6.7082, √2 = 1.4142
2. √67 = 8.1854, √3 = 1.7321
3. √21 = 4.5826, √4 = 2.0000
4. √23 = 4.7958, √5 = 2.2361
5. √2 = 1.4142, √6 = 2.4495

Travailler avec le processeur

Si vous souhaitez exécuter une fonction SQRT () 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 le CPU et retournerons les racines carrées 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 ("Root carré:")
imprimer (torche.SQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Racine carrée:
tenseur ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Fonctionnement:

1. √12 = 3.4641
2. √34 = 5.8310
3. √56 = 7.4833
4. √1 = 1.0000
5. √10 = 3.1623

Exemple 2:

Dans cet exemple, nous créerons un tenseur avec deux dimensions qui a 5 éléments sur le CPU dans chaque ligne et renvoyer la racine carrée 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 ("Root carré:")
imprimer (torche.SQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Racine carrée:
tenseur ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Fonctionnement:

1. √45 = 6.7082, √2 = 1.4142
2. √67 = 8.1854, √3 = 1.7321
3. √21 = 4.5826, √4 = 2.0000
4. √23 = 4.7958, √5 = 2.2361
5. √2 = 1.4142, √6 = 2.4495

Conclusion

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