Pytorch -hstack
Un tenseur est un tableau multidimensionnel utilisé pour stocker des données. Donc, 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.hstack ()
torche.hstack () rejoint deux tenseurs ou plus horizontalement,
Syntaxe:
torche.hstack (Tensor_Object1, Tensor_Object2,…)
Paramètre:
Il faut deux ou plusieurs tenseurs.
Exemple 1:
Dans cet exemple, nous créerons deux tenseurs unidimensionnels et nous les rejoindrons horizontalement à l'aide d'une torche.hstack ().
Importer une torche
#create 2 tenseurs
data1 = torche.tenseur ([10,20,40,50])
data2 = torche.tenseur ([2,3,4,5])
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
#join deux tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tenseur joint: tenseur ([10, 20, 40, 50, 2, 3, 4, 5])
Deux tenseurs ont rejoint horizontalement.
Exemple 2:
Dans cet exemple, nous créerons cinq tenseurs unidimensionnels et nous les rejoindrons horizontalement à l'aide d'une torche.hstack ().
Importer une torche
#Create 5 tenseurs
data1 = torche.tenseur ([10,20,40,50])
data2 = torche.tenseur ([2,3,4,5])
data3 = torche.tenseur ([12])
data4 = torche.tenseur ([100])
data5 = torche.tenseur ([120 456])
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
tenseur ([12])
tenseur ([100])
tenseur ([120, 456])
Tenseur joint: tenseur ([10, 20, 40, 50, 2, 3, 4, 5, 12, 100, 120, 456])
Cinq tenseurs sont rejoints horizontalement.
Exemple 3:
Dans cet exemple, nous créerons cinq tenseurs bidimensionnels et nous les rejoindrons horizontalement à l'aide d'une torche.hstack ().
Importer une torche
#create 5 tenseurs avec 2 dimensions chacun
data1 = torche.Tensor ([[10,20,40,50], [1,2,3,4]]))
data2 = torche.Tensor ([[2,3,4,5], [20,70,89,0]]))
data3 = torche.tenseur ([[12], [56]])
data4 = torche.tenseur ([[100], [67]])
data5 = torche.Tensor ([[120], [456]])
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:Tensor ([[10, 20, 40, 50],
[1, 2, 3, 4]])
Tensor ([[2, 3, 4, 5],
[20, 70, 89, 0]])
tenseur ([[12],
[56]])
tenseur ([[100],
[67]])
Tensor ([[120],
[456]])
Tenseur joint: tenseur ([[10, 20, 40, 50, 2, 3, 4, 5, 12, 100, 120],
[1, 2, 3, 4, 20, 70, 89, 0, 56, 67, 456]]))
Cinq tenseurs sont rejoints horizontalement.
Travailler avec le processeur
Si vous souhaitez exécuter une fonction hstack () sur le CPU, alors nous devons créer un tenseur avec une fonction CPU (). Cela fonctionnera sur une machine CPU.
Pour le moment, lorsque nous créons un tenseur, nous pouvons utiliser la fonction CPU ().
Syntaxe:
torche.tenseur (données).CPU()
Exemple 1:
Dans cet exemple, nous créerons deux tenseurs unidimensionnels sur le CPU et les rejoindrons horizontalement à l'aide d'une torche.hstack ().
Importer une torche
#create 2 tenseurs
data1 = torche.Tensor ([10,20,40,50].CPU()
data2 = torche.Tensor ([2,3,4,5].CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
#join deux tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tenseur joint: tenseur ([10, 20, 40, 50, 2, 3, 4, 5])
Deux tenseurs sont rejoints horizontalement.
Exemple 2:
Dans cet exemple, nous créerons cinq tenseurs unidimensionnels sur le processeur et les joindrons à les horizontalement à l'aide d'une torche.hstack ().
Module de torche #mportImporter une torche
#Create 5 tenseurs
data1 = torche.tenseur ([10,20,40,50]).CPU()
data2 = torche.tenseur ([2,3,4,5]).CPU()
data3 = torche.tenseur ([12]).CPU()
data4 = torche.tenseur ([100]).CPU()
data5 = torche.tenseur ([120 456]).CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
tenseur ([12])
tenseur ([100])
tenseur ([120, 456])
Tenseur joint: tenseur ([10, 20, 40, 50, 2, 3, 4, 5, 12, 100, 120, 456])
Cinq tenseurs sont rejoints horizontalement.
Exemple 3:
Dans cet exemple, nous créerons cinq tenseurs bidimensionnels sur le CPU et les rejoindrons horizontalement à l'aide d'une torche.hstack ().
Importer une torche
#create 5 tenseurs avec 2 dimensions chacun
data1 = torche.Tensor ([[10,20,40,50], [1,2,3,4]])).CPU()
data2 = torche.Tensor ([[2,3,4,5], [20,70,89,0]])).CPU()
data3 = torche.tenseur ([[12], [56]]).CPU()
data4 = torche.tenseur ([[100], [67]]).CPU()
data5 = torche.Tensor ([[120], [456]]).CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.hstack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:Tensor ([[10, 20, 40, 50],
[1, 2, 3, 4]])
Tensor ([[2, 3, 4, 5],
[20, 70, 89, 0]])
tenseur ([[12],
[56]])
tenseur ([[100],
[67]])
Tensor ([[120],
[456]])
Tenseur joint: tenseur ([[10, 20, 40, 50, 2, 3, 4, 5, 12, 100, 120],
[1, 2, 3, 4, 20, 70, 89, 0, 56, 67, 456]]))
Cinq tenseurs sont rejoints horizontalement.
Conclusion
Nous avons vu comment rejoindre deux ou plusieurs tenseurs horizontalement à Pytorch en utilisant la fonction hstack (). Dans cet article, nous avons mis en œuvre plusieurs exemples pour rejoindre un et deux tenseurs de dimension et également implémenté HSTACK () sur le CPU à l'aide de la fonction CPU ().