Pytorch - vstack

Discutiremos sobre o vstack () neste tutorial de Pytorch.

Pytorch é uma estrutura de código aberto disponível com uma linguagem de programação Python. Tensor é uma matriz multidimensional usada para armazenar os dados. Para usar um tensor, temos que importar o módulo da tocha. Para criar um tensor, o método usado é tensor ().

Sintaxe:

tocha.Tensor (dados)

Onde os dados são uma matriz multidimensional.

Tocha.vstack ()

Tocha.vstack () se junta a dois ou mais tensores verticalmente.

Sintaxe:

tocha.vstack (tensor_object1, tensor_object2,…)

Parâmetro:

Leva dois ou mais tensores.

Exemplo 1:

Neste exemplo, criaremos dois tensores unidimensionais e nos juntaremos a eles usando a tocha verticalmente.função vstack ().

#import módulo tocha
importação de tocha
#Crie 2 tensores
Data1 = Torch.Tensor ([10,20,40,50])
Data2 = Torch.tensor ([2,3,4,5])
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
#até dois tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2)))

Saída:

Tensores reais:
tensor ([10, 20, 40, 50])
tensor ([2, 3, 4, 5])
Juntou -se ao tensor: tensor ([[10, 20, 40, 50],
[2, 3, 4, 5]])

Dois tensores são unidos verticalmente.

Exemplo 2:

Neste exemplo, criaremos cinco tensores unidimensionais e nos juntaremos a eles verticalmente usando a tocha.função vstack ().

#import módulo tocha
importação de tocha
#Crie 5 tensores
Data1 = Torch.Tensor ([10,20,40,50])
Data2 = Torch.tensor ([2,3,4,5])
Data3 = Torch.Tensor ([12,45,67,89])
Data4 = Torch.Tensor ([100,32,45,67])
Data5 = Torch.Tensor ([120,456,1,1])
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#até cinco tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2, Data3, Data4, Data5))))

Saída:

Tensores reais:
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])
Juntou -se ao tensor: tensor ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])

Cinco tensores são unidos verticalmente.

Exemplo 3:

Neste exemplo, criaremos cinco tensores bidimensionais e nos juntaremos a eles verticalmente usando a tocha.função vstack ().

#import módulo tocha
importação de tocha
#crete 5 tensores com 2 dimensões cada
Data1 = Torch.tensor ([[10,20,40,50], [1,2,3,4]]))
Data2 = Torch.tensor ([[2,3,4,5], [20,70,89,0]])
Data3 = Torch.tensor ([[12,4,5,6], [56,34,56,787]])
Data4 = Torch.tensor ([[100,1,2,3], [67,87,6,78]])
Data5 = Torch.tensor ([[120,33,56,78], [45,56,78,6]])))
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#até cinco tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2, Data3, Data4, Data5))))

Saída:

Tensores reais:
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]])
Juntou -se ao tensor: 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 são unidos verticalmente.

Trabalhe com a CPU

Se você deseja executar uma função vStack () na CPU, temos que criar um tensor com uma função CPU (). Isso será executado em uma máquina de CPU.

Quando criamos um tensor, desta vez, podemos usar a função CPU ().

Sintaxe:

tocha.Tensor (dados).CPU()

Exemplo 1:

Neste exemplo, criaremos dois tensores unidimensionais na CPU e nos juntaremos a eles verticalmente usando a tocha.função vstack ().

#import módulo tocha
importação de tocha
#Crie 2 tensores
Data1 = Torch.Tensor ([10,20,40,50]).CPU()
Data2 = Torch.tensor ([2,3,4,5]).CPU()
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
#até dois tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2)))

Saída:

Tensores reais:
tensor ([10, 20, 40, 50])
tensor ([2, 3, 4, 5])
Juntou -se ao tensor: tensor ([[10, 20, 40, 50],
[2, 3, 4, 5]])

Dois tensores são unidos verticalmente.

Exemplo 2:

Neste exemplo, criaremos cinco tensores unidimensionais na CPU e nos juntaremos a eles verticalmente usando a tocha.função vstack ().

#import módulo tocha
importação de tocha
#Crie 5 tensores
Data1 = Torch.Tensor ([10,20,40,50]).CPU()
Data2 = Torch.tensor ([2,3,4,5]).CPU()
Data3 = Torch.Tensor ([12,45,67,89]).CPU()
Data4 = Torch.Tensor ([100,32,45,67]).CPU()
Data5 = Torch.Tensor ([120,456,1,1]).CPU()
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#até cinco tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2, Data3, Data4, Data5))))

Saída:

Tensores reais:
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])
Juntou -se ao tensor: tensor ([[10, 20, 40, 50],
[2, 3, 4, 5],
[12, 45, 67, 89],
[100, 32, 45, 67],
[120, 456, 1, 1]])

Cinco tensores são unidos verticalmente.

Exemplo 3:

Neste exemplo, criaremos cinco tensores bidimensionais na CPU e nos juntaremos a eles verticalmente usando a tocha.função vstack ().

#import módulo tocha
importação de tocha
#crete 5 tensores com 2 dimensões cada
Data1 = Torch.tensor ([[10,20,40,50], [1,2,3,4]])).CPU()
Data2 = Torch.tensor ([[2,3,4,5], [20,70,89,0]]).CPU()
Data3 = Torch.tensor ([[12,4,5,6], [56,34,56,787]]).CPU()
Data4 = Torch.tensor ([[100,1,2,3], [67,87,6,78]]).CPU()
Data5 = Torch.tensor ([[120,33,56,78], [45,56,78,6]]))).CPU()
#mostrar
Imprimir ("Tensores reais:")
Imprimir (Data1)
Imprimir (Data2)
Imprimir (Data3)
Imprimir (Data4)
Imprimir (Data5)
#até cinco tensores
impressão ("juntou -se a tensor:", tocha.Vstack ((Data1, Data2, Data3, Data4, Data5))))

Saída:

Tensores reais:
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]])
Juntou -se ao tensor: 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 são unidos verticalmente.

Conclusão

Aprendemos a se juntar a dois ou mais tensores verticalmente em pytorch usando a função vStack (). Neste artigo, implementamos os vários exemplos para ingressar em um e bidimensional tensores e implementamos o vstack () na CPU usando a função CPU ().