Pytorch - cuadrado

Devolveremos el cuadrado de todos los elementos en tensor utilizando el método cuadrado () 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.cuadrado()

Square () en Pytorch devuelve el cuadrado de cada elemento en Pytorch Tensor. Se necesita un parámetro.

Sintaxis:

antorcha.cuadrado (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 el cuadrado 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 ("cuadrado:")
imprimir (antorcha.cuadrado (data1))

Producción:

Tensor real:
Tensor ([12, 34, 56, 1, 10])
Cuadrado:
Tensor ([144, 1156, 3136, 1, 100])

Laboral:

1. 12 * 12 = 144
2. 34 * 34 = 1156
3. 56 * 56 = 3136
4. 1 * 1 = 1
5. 10 * 10 = 100

Ejemplo 2:

En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en cada fila y devolveremos los cuadrados 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 ("cuadrado")
imprimir (antorcha.cuadrado (data1))

Producción:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[-2, 3, 4, 5, 6]])
Cuadrado
Tensor ([[2025, 4489, 441, 529, 4],
[4, 9, 16, 25, 36]])

Laboral:

1. 45 * 45 = 2025, -2 * -2 = 4
2. 67 * 67 = 4489,3 * 3 = 9
3. 21 * 21 = 441,4 * 4 = 16
4. 23 * 23 = 529,5 * 5 = 25
5. 2 * 2 = 4,6 * 6 = 36

Trabajar con CPU

Si desea ejecutar una función cuadrada () 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 función de la CPU y devolveremos el cuadrado 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 ("cuadrado:")
imprimir (antorcha.cuadrado (data1))

Producción:

Tensor real:
Tensor ([12, 34, 56, 1, 10])
Cuadrado:
Tensor ([144, 1156, 3136, 1, 100])

Laboral:

1. 12 * 12 = 144
2. 34 * 34 = 1156
3. 56 * 56 = 3136
4. 1 * 1 = 1
5. 10 * 10 = 100

Ejemplo 2:

En este ejemplo, crearemos un tensor con dos dimensiones que tenga 5 elementos en una función de CPU en cada fila y devolveremos los cuadrados 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 ("cuadrado")
imprimir (antorcha.cuadrado (data1))

Producción:

Tensor real:
Tensor ([[45, 67, 21, 23, 2],
[-2, 3, 4, 5, 6]])
Cuadrado
Tensor ([[2025, 4489, 441, 529, 4],
[4, 9, 16, 25, 36]])

Laboral:

1. 45 * 45 = 2025, -2 * -2 = 4
2. 67 * 67 = 4489,3 * 3 = 9
3. 21 * 21 = 441,4 * 4 = 16
4. 23 * 23 = 529,5 * 5 = 25
5. 2 * 2 = 4,6 * 6 = 36

Conclusión

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