Este artigo exemplificará o método de adição de dimensões a um Tensor no PyTorch.
Como adicionar uma dimensão ao tensor específico no PyTorch?
Os usuários podem adicionar dimensões a qualquer Tensor, como um tensor 1D ou um tensor 2D no PyTorch. Para adicionar novas dimensões aos tensores em uma posição específica, confira os exemplos a seguir para um melhor entendimento:
- Exemplo 1: Adicionar dimensão a um tensor 1D no PyTorch
- Exemplo 2: Adicionar dimensão a um tensor 2D no PyTorch
Exemplo 1: Adicionar dimensão a um tensor 1D no PyTorch
Neste exemplo, criaremos um tensor 1D e adicionaremos uma dimensão a ele em uma posição específica. Siga as etapas listadas abaixo para uma demonstração prática:
Etapa 1: importar biblioteca
Primeiro, importe a biblioteca da tocha:
importar tocha
Etapa 2: criar tensor 1D
Em seguida, crie um tensor unidimensional. Por exemplo, criamos o seguinte tensor e o armazenamos no “ x ' variável:
x = tocha. tensor ( [ 5 , 3 , 8 , 2 ] )
Etapa 3: visualizar a forma do tensor
A seguir, exiba a forma do tensor recém-criada para visualizar suas dimensões:
A saída abaixo indica que nosso tensor é unidimensional:
Etapa 4: adicionar dimensão ao tensor 1D
Agora, utilize o “ torch.unsqueeze(entrada, escurecimento) ”Função para adicionar a dimensão ao tensor 1D na posição específica. Por exemplo, estamos adicionando a dimensão ao tensor no índice 0:
Aqui,
- “ novas_tens ”é a variável que inclui a dimensão adicionada.
- “ x ”É o tensor de entrada.
- “ escuro = 0 ”É usado para adicionar dimensão no índice 0.
Etapa 5: verificar a saída
Por fim, certifique-se de que uma nova dimensão foi adicionada ao tensor ou não:
Na saída abaixo, pode-se observar que a nova dimensão foi adicionada ao tensor 1D no índice 0:
Além disso, os usuários também podem adicionar dimensões a outras posições. Aqui, adicionamos a dimensão no primeiro índice:
Exemplo 2: Adicionar dimensão a um tensor 2D no PyTorch
Aqui, criaremos/faremos um tensor 2D e adicionaremos uma dimensão a ele na posição específica. Experimente as etapas fornecidas para implementação prática:
Etapa 1: importar biblioteca do Torch
Primeiro, importe a biblioteca da tocha:
Etapa 2: crie um tensor 2D
Em seguida, crie um tensor bidimensional. Por exemplo, criamos o seguinte tensor e o armazenamos no “ x ' variável:
Etapa 3: visualizar a forma do tensor
Depois disso, exiba a forma do tensor recém-criada para visualizar suas dimensões:
De acordo com a saída abaixo, este tensor é bidimensional:
Etapa 4: adicionar dimensão ao tensor 2D
Agora, adicione a dimensão ao tensor 2D na posição específica usando o “torch.unsqueeze (entrada, escurecimento)” função. Por exemplo, estamos adicionando a dimensão ao tensor no índice 0:
Etapa 5: verificar a saída
Por último, verifique se a nova dimensão foi adicionada ao tensor 2D ou não:
A saída abaixo indica que a nova dimensão foi adicionada com sucesso ao tensor 2D no índice 0:
Observação : Você pode acessar nosso Google Colab Notebook neste link .
Explicamos com eficiência o método de adicionar dimensão a um tensor no PyTorch usando diferentes exemplos.
Conclusão
Para adicionar dimensões a um tensor no PyTorch, primeiro importe a biblioteca PyTorch. Em seguida, crie um tensor 1D ou 2D e visualize suas dimensões. Depois disso, adicione a dimensão a um tensor na posição específica usando o “ torch.unsqueeze(entrada, escurecimento) ”função. Os usuários precisam passar o tensor de entrada e a posição do índice desejada como parâmetro para esta função. Este artigo exemplificou o método de adição de dimensões a um tensor no PyTorch.