Pytorch - rsqrt

Nous retournerons la racine carrée réciproque de tous les éléments du tenseur à l'aide de 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.rsqrt ()

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

Syntaxe:

torche.RSQRT (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 réciproques 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 ("Racine carrée réciproque:")
imprimer (torche.RSQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Racine carrée réciproque:
Tensor ([0.2887, 0.1715, 0.1336, 1.0000, 0.3162])

Fonctionnement:

1. 1 / √12 = 0.2887
2. 1 / √34 = 0.1715
3. 1 / √56 = 0.1336
4. 1 / √1 = 1.0000
5. 1 / √10 = 0.3162

Exemple 2:

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

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Racine carrée réciproque:
tenseur ([[0.1491, 0.1222, 0.2182, 0.2085, 0.7071],
[0.7071, 0.5774, 0.5000, 0.4472, 0.4082]])

Fonctionnement:

1. 1 / √45 = 0.1491, 1 / √2 = 0.7071
2. 1 / √67 = 0.1222, 1 / √3 = 0.5774
3. 1 / √21 = 0.2182, 1 / √4 = 0.5000
4. 1 / √23 = 0.2085, 1 / √5 = 0.4472
5. 1 / √2 = 0.7071, 1 / √6 = 0.4082

Travailler avec le processeur

Si vous souhaitez exécuter une fonction RSQRT () 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 réciproques 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 ("Racine carrée réciproque:")
imprimer (torche.RSQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([12, 34, 56, 1, 10])
Racine carrée réciproque:
Tensor ([0.2887, 0.1715, 0.1336, 1.0000, 0.3162])

Fonctionnement:

1. 1 / √12 = 0.2887
2. 1 / √34 = 0.1715
3. 1 / √56 = 0.1336
4. 1 / √1 = 1.0000
5. 1 / √10 = 0.3162

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 réciproque 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 ("Racine carrée réciproque:")
imprimer (torche.RSQRT (DATA1))

Sortir:

Tensor réel:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Racine carrée réciproque:
tenseur ([[0.1491, 0.1222, 0.2182, 0.2085, 0.7071],
[0.7071, 0.5774, 0.5000, 0.4472, 0.4082]])

Fonctionnement:

1. 1 / √45 = 0.1491, 1 / √2 = 0.7071
2. 1 / √67 = 0.1222, 1 / √3 = 0.5774
3. 1 / √21 = 0.2182, 1 / √4 = 0.5000
4. 1 / √23 = 0.2085, 1 / √5 = 0.4472
5. 1 / √2 = 0.7071, 1 / √6 = 0.4082

Conclusion

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