Como usar a operação de expansão no PyTorch?

PyTorch é uma biblioteca de aprendizado de máquina que permite aos usuários trabalhar com tensores. Tensores são estruturas de dados essenciais que podem ter diferentes tipos de dados e formas. Às vezes, os usuários podem querer realizar a operação de expansão em tensores para ampliar seu tamanho. A operação de expansão permite aos usuários repetir um tensor ao longo de dimensões especificadas. PyTorch fornece o atributo “expand()” que usa um tensor e uma lista de tamanhos como entradas. Ele retorna um novo tensor que possui os mesmos dados, mas com dimensões diferentes.

Este artigo ilustrará o método para usar a operação de expansão em tensores no PyTorch.

Como usar a operação de expansão no PyTorch?

Para usar a operação de expansão no PyTorch, verifique as etapas fornecidas abaixo:

• Importar biblioteca PyTorch
• Crie um tensor desejado
• Ver o tamanho do tensor de entrada
• Expanda o tensor usando o “ expandir() ”atributo
• Exibir tensor expandido e seu tamanho

Etapa 1: importar biblioteca PyTorch

Primeiro, importe o “ tocha ”biblioteca para usar a operação de expansão:

Etapa 2: crie um tensor

Em seguida, crie um tensor desejado usando o “ tocha.tensor() ”Funciona e imprime seus elementos. Aqui, estamos criando o seguinte “ Dezenas ” tensor:

Na saída abaixo, o tensor foi criado com sucesso:

Etapa 3: visualizar o tamanho do tensor de entrada

A seguir, veja o tamanho do “ Dezenas ”tensor usando o“ tamanho() ”atributo:

De acordo com a saída abaixo, o tamanho do tensor é 3×1:

Etapa 4: expandir o tensor

Agora, use o “ expandir() ”Atributo para realizar a operação de expansão e estender o tensor para uma nova dimensão. É necessário fornecer o tensor e a lista de tamanhos como entrada. Aqui, estamos expandindo o tensor para o tamanho 3×4:

Etapa 5: exibir o tensor expandido e seu tamanho

Por último, imprima os elementos do tensor expandido e seu tamanho:

A saída abaixo mostra o tensor expandido e seu tamanho, ou seja, 3×4. Isto indica que a operação expandida foi executada com sucesso:

Explicamos o método eficiente para usar a operação de expansão no PyTorch

Observação : Você pode acessar nosso Google Colab Notebook neste link .

Conclusão

Para usar a operação de expansão no PyTorch, primeiro importe a biblioteca da tocha. Em seguida, crie o tensor desejado e visualize seus elementos e tamanho. Em seguida, use o “ expandir() ”Atributo para expandir o tensor de entrada. Por último, imprima o tensor expandido e veja seu tamanho. Este artigo demonstrou o método para usar a operação de expansão em tensores no PyTorch.