Pytorch - vstack
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.vStack ()
Torche.Vstack () rejoint deux tenseurs ou plus verticalement.
Syntaxe:
torche.Vstack (Tensor_Object1, Tensor_Object2,…)Paramètre:
Il faut deux ou plusieurs tenseurs.
Exemple 1:
Dans cet exemple, nous créerons deux tenseurs unidimensionnels et les rejoindrons verticalement en utilisant le flambeau.fonction vStack ().
Module de torche #mportImporter 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.vStack ((data1, data2)))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[2, 3, 4, 5]])
Deux tenseurs sont rejoints verticalement.
Exemple 2:
Dans cet exemple, nous créerons cinq tenseurs unidimensionnels et les rejoindrons verticalement en utilisant le flambeau.fonction vStack ().
Module de torche #mportImporter une torche
#Create 5 tenseurs
data1 = torche.tenseur ([10,20,40,50])
data2 = torche.tenseur ([2,3,4,5])
data3 = torche.Tensor ([12,45,67,89])
data4 = torche.Tensor ([100,32,45,67])
data5 = torche.Tensor ([120 456,1,1])
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.vStack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tensor ([12, 45, 67, 89])
tenseur ([100, 32, 45, 67])
tenseur ([120, 456, 1, 1])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])
Cinq tenseurs sont rejoints verticalement.
Exemple 3:
Dans cet exemple, nous créerons cinq tenseurs bidimensionnels et les rejoindrons verticalement en utilisant le flambeau.fonction vStack ().
Module de torche #mportImporter 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.Tensor ([[12,4,5,6], [56,34,56,787]]))
data4 = torche.Tensor ([[100,1,2,3], [67,87,6,78]])
data5 = torche.Tensor ([[120,33,56,78], [45,56,78,6]])
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.vStack ((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]])
Tensor ([[12, 4, 5, 6],
[56, 34, 56, 787]])
tenseur ([[100, 1, 2, 3],
[67, 87, 6, 78]])
Tensor ([[120, 33, 56, 78],
[45, 56, 78, 6]])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[1, 2, 3, 4],
[2, 3, 4, 5],
[20, 70, 89, 0],
[12, 4, 5, 6],
[56, 34, 56, 787],
[100, 1, 2, 3],
[67, 87, 6, 78],
[120, 33, 56, 78],
[45, 56, 78, 6]])
Cinq tenseurs sont rejoints verticalement.
Travailler avec le processeur
Si vous souhaitez exécuter une fonction VStack () 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 deux tenseurs unidimensionnels sur le CPU et les rejoindrons verticalement en utilisant la torche.fonction vStack ().
Module de torche #mportImporter une torche
#create 2 tenseurs
data1 = torche.tenseur ([10,20,40,50]).CPU()
data2 = torche.tenseur ([2,3,4,5]).CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
#join deux tenseurs
Print ("Tensor joint:", torche.vStack ((data1, data2)))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[2, 3, 4, 5]])
Deux tenseurs sont rejoints verticalement.
Exemple 2:
Dans cet exemple, nous créerons cinq tenseurs unidimensionnels sur le CPU et les rejoindrons verticalement en utilisant la torche.fonction vStack ().
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.Tensor ([12,45,67,89]).CPU()
data4 = torche.Tensor ([100,32,45,67]).CPU()
data5 = torche.Tensor ([120 456,1,1]).CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.vStack ((data1, data2, data3, data4, data5))))
Sortir:
Tenseurs réels:tenseur ([10, 20, 40, 50])
tenseur ([2, 3, 4, 5])
Tensor ([12, 45, 67, 89])
tenseur ([100, 32, 45, 67])
tenseur ([120, 456, 1, 1])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])
Cinq tenseurs sont rejoints verticalement.
Exemple 3:
Dans cet exemple, nous créerons cinq tenseurs bidimensionnels sur le CPU et les rejoindrons en utilisant la torche.fonction vStack ().
Module de torche #mportImporter 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.Tensor ([[12,4,5,6], [56,34,56,787]])).CPU()
data4 = torche.Tensor ([[100,1,2,3], [67,87,6,78]]).CPU()
data5 = torche.Tensor ([[120,33,56,78], [45,56,78,6]]).CPU()
#afficher
Imprimer ("Tenseurs réels:")
Imprimer (DATA1)
Imprimer (data2)
Imprimer (DATA3)
Imprimer (Data4)
Imprimer (DATA5)
#join cinq tenseurs
Print ("Tensor joint:", torche.vStack ((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]])
Tensor ([[12, 4, 5, 6],
[56, 34, 56, 787]])
tenseur ([[100, 1, 2, 3],
[67, 87, 6, 78]])
Tensor ([[120, 33, 56, 78],
[45, 56, 78, 6]])
Tensor rejoint: tenseur ([[10, 20, 40, 50],
[1, 2, 3, 4],
[2, 3, 4, 5],
[20, 70, 89, 0],
[12, 4, 5, 6],
[56, 34, 56, 787],
[100, 1, 2, 3],
[67, 87, 6, 78],
[120, 33, 56, 78],
[45, 56, 78, 6]])
Cinq tenseurs sont rejoints verticalement.
Conclusion
Nous avons appris à rejoindre deux ou plusieurs tenseurs verticalement dans Pytorch en utilisant la fonction VStack (). Dans cet article, nous avons mis en œuvre les plusieurs exemples pour rejoindre un et deux tenseurs dimensionnels et mis en œuvre le VStack () sur CPU à l'aide de la fonction CPU ().