Pytorch - SQRT

Devolveremos la raíz cuadrada de todos los elementos en el tensor utilizando el método sqrt () en este tutorial de pytorch.

Pytorch es un marco de código abierto disponible con un lenguaje de programación de Python. Tensor es una matriz multidimensional que se utiliza para almacenar los datos. Para usar un tensor, tenemos que importar el módulo de antorcha. Para crear un tensor, el método utilizado es tensor ().

Sintaxis:

antorcha.Tensor (datos)

Donde los datos son una matriz multidimensional.

Antorcha.sqrt ()

Sqrt () en pytorch devuelve la raíz cuadrada de cada elemento en el tensor de pytorch. Se necesita un parámetro.

Sintaxis:

antorcha.sqrt (tensor_object)

Parámetro:

tensor_object es un tensor

Ejemplo 1:

En este ejemplo, crearemos un tensor con una dimensión que tenga 5 elementos y devuelva las raíces cuadradas de estos 5 elementos en un tensor.

#módulo de antorcha de Import
antorcha de importación
#cree un tensor
datos1 = antorcha.tensor ([12,34,56,1,10])
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("Raíz cuadrada:")
imprimir (antorcha.sqrt (data1))

Producción:

Tensor real:
Tensor ([12, 34, 56, 1, 10])
Raíz cuadrada:
tensor ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Laboral:

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

Ejemplo 2:

En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en cada fila y devolveremos la raíz cuadrada de los elementos.

#módulo de antorcha de Import
antorcha de importación
#cree un tensor 2D
datos1 = antorcha.Tensor ([[45,67,21,23,2], [2,3,4,5,6]])
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("Raíz cuadrada:")
imprimir (antorcha.sqrt (data1))

Producción:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raíz cuadrada:
tensor ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Laboral:

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

Trabajar con CPU

Si desea ejecutar una función sqrt () en la CPU, tenemos que crear un tensor con una función CPU (). Esto se ejecutará en una máquina CPU.

Cuando creamos un tensor, esta vez, podemos usar la función CPU ().

Sintaxis:

antorcha.Tensor (datos).UPC()

Ejemplo 1:

En este ejemplo, crearemos un tensor con una dimensión que tenga 5 elementos en la CPU y devolveremos las raíces cuadradas de estos 5 elementos en un tensor.

#módulo de antorcha de Import
antorcha de importación
#cree un tensor
datos1 = antorcha.tensor ([12,34,56,1,10]).UPC()
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("Raíz cuadrada:")
imprimir (antorcha.sqrt (data1))

Producción:

Tensor real:
Tensor ([12, 34, 56, 1, 10])
Raíz cuadrada:
tensor ([3.4641, 5.8310, 7.4833, 1.0000, 3.1623])

Laboral:

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

Ejemplo 2:

En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en la CPU en cada fila y devolveremos la raíz cuadrada de los elementos.

#módulo de antorcha de Import
antorcha de importación
#cree un tensor 2D
datos1 = antorcha.Tensor ([[45,67,21,23,2], [2,3,4,5,6]]).UPC()
#mostrar
Imprimir ("Tensor real:")
Imprimir (Data1)
Imprimir ("Raíz cuadrada:")
imprimir (antorcha.sqrt (data1))

Producción:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[2, 3, 4, 5, 6]])
Raíz cuadrada:
tensor ([[6.7082, 8.1854, 4.5826, 4.7958, 1.4142],
[1.4142, 1.7321, 2.0000, 2.2361, 2.4495]])

Laboral:

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

Conclusión

En esta lección de Pytorch, discutimos sobre la función sqrt (). Devuelve la raíz cuadrada de cada elemento en el tensor Pytorch. Discutimos los dos ejemplos con los diferentes tensores dimensionales para realizar la función sqrt ().