Pytorch - Sqrt

Retornaremos a raiz quadrada de todos os elementos no tensor usando o método sqrt () 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.sqrt ()

Sqrt () em pytorch retorna a raiz quadrada de todos os elementos no tensor de pytorch. É preciso um parâmetro.

Sintaxe:

tocha.sqrt (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 as raízes quadradas desses 5 elementos em um tensor.

#import módulo tocha
importação de tocha
#Crie um tensor
Data1 = Torch.Tensor ([12,34,56,1,10])
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("raiz quadrada:")
Imprimir (tocha.SQRT (Data1))

Saída:

Tensor real:
tensor ([12, 34, 56, 1, 10])
Raiz quadrada:
Tensor ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Trabalhando:

1. √12 = 3.4641
2. √34 = 5.8310
3. √56 = 7.4833
4. √1 = 1.0000
5. √10 = 3.1623

Exemplo 2:

Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos em cada linha e devolveremos a raiz quadrada dos elementos.

#import módulo tocha
importação de tocha
#Crie um tensor 2D
Data1 = Torch.tensor ([[45,67,21,23,2], [2,3,4,5,6]])))
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("raiz quadrada:")
Imprimir (tocha.SQRT (Data1))

Saída:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raiz quadrada:
Tensor ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Trabalhando:

1. √45 = 6.7082, √2 = 1.4142
2. √67 = 8.1854, √3 = 1.7321
3. √21 = 4.5826, √4 = 2.0000
4. √23 = 4.7958, √5 = 2.2361
5. √2 = 1.4142, √6 = 2.4495

Trabalhe com a CPU

Se você deseja executar uma função SQRT () 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 CPU e retornaremos as raízes quadradas desses 5 elementos em um tensor.

#import módulo tocha
importação de tocha
#Crie um tensor
Data1 = Torch.Tensor ([12,34,56,1,10]).CPU()
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("raiz quadrada:")
Imprimir (tocha.SQRT (Data1))

Saída:

Tensor real:
tensor ([12, 34, 56, 1, 10])
Raiz quadrada:
Tensor ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Trabalhando:

1. √12 = 3.4641
2. √34 = 5.8310
3. √56 = 7.4833
4. √1 = 1.0000
5. √10 = 3.1623

Exemplo 2:

Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos na CPU em cada linha e devolva a raiz quadrada dos elementos.

#import módulo tocha
importação de tocha
#Crie um tensor 2D
Data1 = Torch.tensor ([[45,67,21,23,2], [2,3,4,5,6]]))).CPU()
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("raiz quadrada:")
Imprimir (tocha.SQRT (Data1))

Saída:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raiz quadrada:
Tensor ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Trabalhando:

1. √45 = 6.7082, √2 = 1.4142
2. √67 = 8.1854, √3 = 1.7321
3. √21 = 4.5826, √4 = 2.0000
4. √23 = 4.7958, √5 = 2.2361
5. √2 = 1.4142, √6 = 2.4495

Conclusão

Nesta lição de Pytorch, discutimos sobre a função SQRT (). Retorna a raiz quadrada de todos os elementos no tensor pytorch. Discutimos os dois exemplos com os diferentes tensores dimensionais para executar a função SQRT ().