Discutiremos sobre el vstack () en este tutorial de pytorch.
Pytorch es un marco de código abierto disponible con un lenguaje de programación de Python. Tensor es una matriz multidimensional que se utiliza para almacenar los datos. Para usar un tensor, tenemos que importar el módulo de antorcha. Para crear un tensor, el método utilizado es tensor ().
Sintaxis:
antorcha.Tensor (datos)
Donde los datos son una matriz multidimensional.
Antorcha.Vstack ()
Antorcha.vstack () se une a dos o más tensores verticalmente.
Sintaxis:
antorcha.vstack (tensor_object1, tensor_object2, ...)
Parámetro:
Se necesitan dos o más tensores.
Ejemplo 1:
En este ejemplo, crearemos dos tensores unidimensionales y nos uniremos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 2 tensores
datos1 = antorcha.Tensor ([10,20,40,50])
data2 = antorcha.tensor ([2,3,4,5])
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
#unir dos tensores
Imprimir ("unido tensor:", antorcha.vstack ((data1, data2)))
Producción:
Tensores reales:
Tensor ([10, 20, 40, 50])
Tensor ([2, 3, 4, 5])
Tensor unido: Tensor ([[10, 20, 40, 50],
[2, 3, 4, 5]])
Dos tensores se unen verticalmente.
Ejemplo 2:
En este ejemplo, crearemos cinco tensores unidimensionales y nos uniremos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 5 tensores
datos1 = antorcha.Tensor ([10,20,40,50])
data2 = antorcha.tensor ([2,3,4,5])
datos3 = antorcha.tensor ([12,45,67,89])
data4 = antorcha.tensor ([100,32,45,67])
Data5 = antorcha.Tensor ([120,456,1,1])
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#unir a cinco tensores
Imprimir ("unido tensor:", antorcha.Vstack ((Data1, Data2, Data3, Data4, Data5)))
Producción:
Tensores reales:
Tensor ([10, 20, 40, 50])
Tensor ([2, 3, 4, 5])
Tensor ([12, 45, 67, 89])
Tensor ([100, 32, 45, 67])
Tensor ([120, 456, 1, 1])
Tensor unido: Tensor ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])
Cinco tensores se unen verticalmente.
Ejemplo 3:
En este ejemplo, crearemos cinco tensores bidimensionales y nos uniremos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 5 tensores con 2 dimensiones cada uno
datos1 = antorcha.Tensor ([[10,20,40,50], [1,2,3,4]])
data2 = antorcha.Tensor ([[2,3,4,5], [20,70,89,0]]))
datos3 = antorcha.Tensor ([[12,4,5,6], [56,34,56,787]]))
data4 = antorcha.Tensor ([[100,1,2,3], [67,87,6,78]]))
Data5 = antorcha.Tensor ([[120,33,56,78], [45,56,78,6]])
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#unir a cinco tensores
Imprimir ("unido tensor:", antorcha.Vstack ((Data1, Data2, Data3, Data4, Data5)))
Producción:
Tensores reales:
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]])
Tensor ([100, 1, 2, 3],
[67, 87, 6, 78]])
Tensor ([[120, 33, 56, 78],
[45, 56, 78, 6]])
Tensor unido: Tensor ([[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]])
Cinco tensores se unen verticalmente.
Trabajar con CPU
Si desea ejecutar una función vstack () en la CPU, tenemos que crear un tensor con una función CPU (). Esto se ejecutará en una máquina CPU.
Cuando creamos un tensor, esta vez, podemos usar la función CPU ().
Sintaxis:
antorcha.Tensor (datos).UPC()
Ejemplo 1:
En este ejemplo, crearemos dos tensores unidimensionales en la CPU y nos uniremos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 2 tensores
datos1 = antorcha.Tensor ([10,20,40,50]).UPC()
data2 = antorcha.tensor ([2,3,4,5]).UPC()
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
#unir dos tensores
Imprimir ("unido tensor:", antorcha.vstack ((data1, data2)))
Producción:
Tensores reales:
Tensor ([10, 20, 40, 50])
Tensor ([2, 3, 4, 5])
Tensor unido: Tensor ([[10, 20, 40, 50],
[2, 3, 4, 5]])
Dos tensores se unen verticalmente.
Ejemplo 2:
En este ejemplo, crearemos cinco tensores unidimensionales en la CPU y nos uniremos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 5 tensores
datos1 = antorcha.Tensor ([10,20,40,50]).UPC()
data2 = antorcha.tensor ([2,3,4,5]).UPC()
datos3 = antorcha.tensor ([12,45,67,89]).UPC()
data4 = antorcha.tensor ([100,32,45,67]).UPC()
Data5 = antorcha.Tensor ([120,456,1,1]).UPC()
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#unir a cinco tensores
Imprimir ("unido tensor:", antorcha.Vstack ((Data1, Data2, Data3, Data4, Data5)))
Producción:
Tensores reales:
Tensor ([10, 20, 40, 50])
Tensor ([2, 3, 4, 5])
Tensor ([12, 45, 67, 89])
Tensor ([100, 32, 45, 67])
Tensor ([120, 456, 1, 1])
Tensor unido: Tensor ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])
Cinco tensores se unen verticalmente.
Ejemplo 3:
En este ejemplo, crearemos cinco tensores bidimensionales en la CPU y nos uniremos a ellos verticalmente usando la antorcha.función vstack ().
#módulo de antorcha de Import
antorcha de importación
#cree 5 tensores con 2 dimensiones cada uno
datos1 = antorcha.Tensor ([[10,20,40,50], [1,2,3,4]]).UPC()
data2 = antorcha.Tensor ([[2,3,4,5], [20,70,89,0]])).UPC()
datos3 = antorcha.Tensor ([[12,4,5,6], [56,34,56,787]])).UPC()
data4 = antorcha.Tensor ([[100,1,2,3], [67,87,6,78]])).UPC()
Data5 = antorcha.Tensor ([[120,33,56,78], [45,56,78,6]]).UPC()
#mostrar
Imprimir ("Tensores reales:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#unir a cinco tensores
Imprimir ("unido tensor:", antorcha.Vstack ((Data1, Data2, Data3, Data4, Data5)))
Producción:
Tensores reales:
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]])
Tensor ([100, 1, 2, 3],
[67, 87, 6, 78]])
Tensor ([[120, 33, 56, 78],
[45, 56, 78, 6]])
Tensor unido: Tensor ([[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]])
Cinco tensores se unen verticalmente.
Conclusión
Aprendimos a unir dos o más tensores verticalmente en Pytorch usando la función vstack (). En este artículo, implementamos varios ejemplos para unir tensores de uno y dos dimensiones e implementamos el vstack () en CPU utilizando la función CPU ().