Pytorch - rsqrt

Retornaremos a raiz quadrada recíproca de todos os elementos do 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.rsqrt ()

SQRT () em Pytorch retorna a raiz quadrada recíproca de todos os elementos no tensor de pytorch. É preciso um parâmetro.

Sintaxe:

tocha.rsqrt (tensor_object)

Parâmetro:

tensor_object é um tensor

Exemplo 1:

Neste exemplo, criaremos um tensor com uma dimensão que tenha 5 elementos e retorneemos as raízes quadradas recíprocas 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)
Print ("raiz quadrada recíproca:")
Imprimir (tocha.RSQRT (Data1))

Saída:

Tensor real:
tensor ([12, 34, 56, 1, 10])
Raiz quadrada recíproca:
tensor ([0.2887, 0.1715, 0.1336, 1.0000, 0.3162])

Trabalhando:

1. 1/√12 = 0.2887
2. 1/√34 = 0.1715
3. 1/√56 = 0.1336
4. 1/√1 = 1.0000
5. 1/√10 = 0.3162

Exemplo 2:

Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos em cada linha e retornem a raiz quadrada recíproca 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)
Print ("raiz quadrada recíproca:")
Imprimir (tocha.RSQRT (Data1))

Saída:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raiz quadrada recíproca:
tensor ([[0.1491, 0.1222, 0.2182, 0.2085, 0.7071],
[0.7071, 0.5774, 0.5000, 0.4472, 0.4082]])

Trabalhando:

1. 1/√45 = 0.1491, 1/√2 = 0.7071
2. 1/√67 = 0.1222, 1/√3 = 0.5774
3. 1/√21 = 0.2182, 1/√4 = 0.5000
4. 1/√23 = 0.2085, 1/√5 = 0.4472
5. 1/√2 = 0.7071, 1/√6 = 0.4082

Trabalhe com a CPU

Se você deseja executar uma função rsqrt () 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 retorneemos as raízes quadradas recíprocas 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)
Print ("raiz quadrada recíproca:")
Imprimir (tocha.RSQRT (Data1))

Saída:

Tensor real:
tensor ([12, 34, 56, 1, 10])
Raiz quadrada recíproca:
tensor ([0.2887, 0.1715, 0.1336, 1.0000, 0.3162])

Trabalhando:

1. 1/√12 = 0.2887
2. 1/√34 = 0.1715
3. 1/√56 = 0.1336
4. 1/√1 = 1.0000
5. 1/√10 = 0.3162

Exemplo 2:

Neste exemplo, criaremos um tensor com duas dimensões que possuem 5 elementos na CPU em cada linha e retornem a raiz quadrada recíproca 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)
Print ("raiz quadrada recíproca:")
Imprimir (tocha.RSQRT (Data1))

Saída:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raiz quadrada recíproca:
tensor ([[0.1491, 0.1222, 0.2182, 0.2085, 0.7071],
[0.7071, 0.5774, 0.5000, 0.4472, 0.4082]])

Trabalhando:

1. 1/√45 = 0.1491, 1/√2 = 0.7071
2. 1/√67 = 0.1222, 1/√3 = 0.5774
3. 1/√21 = 0.2182, 1/√4 = 0.5000
4. 1/√23 = 0.2085, 1/√5 = 0.4472
5. 1/√2 = 0.7071, 1/√6 = 0.4082

Conclusão

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