Pytorch - Trace

Veremos cómo encontrar el rastro de una matriz tenor dada 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.rastro()

La traza se calcula como la suma de elementos diagonales en una matriz.

Sintaxis:

antorcha.Trace (tensor_object)

Parámetro:

Se necesita un objeto tensor como un parámetro que es un tensor 2D.

Ejemplo 1:

Creemos un tensor que tenga 4 filas y 4 columnas y devuelvamos el rastro de la matriz tensor.

#Importar el módulo de antorcha
antorcha de importación
#cree una matriz tensor 2D
datos1 = antorcha.Tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]])
#mostrar
Imprimir ("Matriz tensor real:")
Imprimir (Data1)
Imprimir ("Trace de una matriz:")
#rastreado
imprimir (antorcha.traza (data1))

Producción:

Matriz tensor real:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Rastro de una matriz:
tensor (8)

La suma de diagonales es: 2+1+5+0 = 8. Por lo tanto, el rastro es 8.

Ejemplo 2:

Creemos un tensor que tenga 2 filas y 2 columnas y devolvamos el rastro de la matriz de tensor.

# Importar el módulo de antorcha
antorcha de importación
#cree una matriz tensor 2D
datos1 = antorcha.Tensor ([[2,33], [3,1]])
#mostrar
Imprimir ("Matriz tensor real:")
Imprimir (Data1)
Imprimir ("Trace de una matriz:")
#rastreado
imprimir (antorcha.traza (data1))

Producción:

Matriz tensor real:
Tensor ([[2, 33],
[3, 1]])
Rastro de una matriz:
tensor (3)

La suma de diagonales es: 2+1 = 3. Por lo tanto, el rastro es 3.

Trabajar con CPU

Si desea ejecutar la función Trace () 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:

Creemos un tensor que tenga 4 filas y 4 columnas en la CPU y devolvamos el rastro de la matriz de tensor.

#Importar el módulo de antorcha
antorcha de importación
#cree una matriz tensor 2D
datos1 = antorcha.Tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]]).UPC()
#mostrar
Imprimir ("Matriz tensor real:")
Imprimir (Data1)
Imprimir ("Trace de una matriz:")
#rastreado
imprimir (antorcha.traza (data1))

Producción:

Matriz tensor real:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Rastro de una matriz:
tensor (8)

La suma de diagonales es: 2+1+5+0 = 8. Por lo tanto, el rastro es 8.

Ejemplo 2:

Creemos un tensor que tenga 2 filas y 2 columnas en la CPU y devolvamos el rastro de la matriz de tensor.

# Importar el módulo de antorcha
antorcha de importación
#cree una matriz tensor 2D
datos1 = antorcha.Tensor ([[2,33], [3,1]]).UPC()
#mostrar
Imprimir ("Matriz tensor real:")
Imprimir (Data1)
Imprimir ("Trace de una matriz:")
#rastreado
imprimir (antorcha.traza (data1))

Producción:

Matriz tensor real:
Tensor ([[2, 33],
[3, 1]])
Rastro de una matriz:
tensor (3)

La suma de diagonales es: 2+1 = 3. Por lo tanto, el rastro es 3.

Conclusión

En esta lección de Pytorch, discutimos sobre la función traza (). Devuelve la suma de elementos diagonales en una matriz. También discutimos los diferentes ejemplos y trabajamos estos ejemplos en una máquina de CPU.