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.quadrado()
Square () em Pytorch retorna o quadrado de todos os elementos em Pytorch Tensor. É preciso um parâmetro.
Sintaxe:
tocha.quadrado (tensor_object)Parâmetro:
tensor_object é um tensor
Exemplo 1:
Neste exemplo, criaremos um tensor com uma dimensão que tenha 5 elementos e retorne o quadrado desses 5 elementos em um tensor.
#import módulo tochaSaída:
Tensor real:Trabalhando:
Exemplo 2:
Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos em cada linha e retornem os quadrados dos elementos.
#import módulo tochaSaída:
Tensor real:Trabalhando:
Trabalhe com a CPU
Se você deseja executar uma função Square () 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 um tensor com uma dimensão que tenha 5 elementos na função da CPU e retorne o quadrado desses 5 elementos em um tensor.
#import módulo tochaSaída:
Tensor real:Trabalhando:
Exemplo 2:
Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos em uma função da CPU em cada linha e retorne os quadrados dos elementos.
#import módulo tochaSaída:
Tensor real:Trabalhando:
Conclusão
Nesta lição de Pytorch, discutimos sobre a função Square (). Retorna o quadrado de todos os elementos no tensor pytorch. Discutimos os dois exemplos com os diferentes tensores dimensionais para executar a função Square ().