Pytorch - Trace

Veremos como encontrar o traço de uma determinada matriz de tenor 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.vestígio()

Trace é calculada como a soma dos elementos diagonais em uma matriz.

Sintaxe:

tocha.traço (tensor_object)

Parâmetro:

É preciso um objeto tensor como um parâmetro que é um tensor 2D.

Exemplo 1:

Vamos criar um tensor que tenha 4 linhas e 4 colunas e retorne o traço da matriz de tensor.

#Vamos importar módulo de tocha
importação de tocha
#Crie uma matriz de tensor 2D
Data1 = Torch.tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]]))
#mostrar
Print ("Matriz Tensor Real:")
Imprimir (Data1)
Print ("Rastreio de uma matriz:")
#return rastreio
Imprimir (tocha.rastreamento (Data1))

Saída:

Tensor Matrix real:
tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Traço de uma matriz:
Tensor (8)

A soma dos diagonais é: 2+1+5+0 = 8. Portanto, o traço é 8.

Exemplo 2:

Vamos criar um tensor que tenha 2 linhas e 2 colunas e retorne o traço da matriz de tensor.

# Vamos importar módulo de tocha
importação de tocha
#Crie uma matriz de tensor 2D
Data1 = Torch.Tensor ([[2,33], [3,1]])
#mostrar
Print ("Matriz Tensor Real:")
Imprimir (Data1)
Print ("Rastreio de uma matriz:")
#return rastreio
Imprimir (tocha.rastreamento (Data1))

Saída:

Tensor Matrix real:
Tensor ([[2, 33],
[3, 1]])
Traço de uma matriz:
Tensor (3)

A soma das diagonais é: 2+1 = 3. Portanto, o traço é 3.

Trabalhe com a CPU

Se você deseja executar a função Trace () 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:

Vamos criar um tensor que tenha 4 linhas e 4 colunas na CPU e retorne o traço da matriz de tensor.

#Vamos importar módulo de tocha
importação de tocha
#Crie uma matriz de tensor 2D
Data1 = Torch.tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]])).CPU()
#mostrar
Print ("Matriz Tensor Real:")
Imprimir (Data1)
Print ("Rastreio de uma matriz:")
#return rastreio
Imprimir (tocha.rastreamento (Data1))

Saída:

Tensor Matrix real:
tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Traço de uma matriz:
Tensor (8)

A soma dos diagonais é: 2+1+5+0 = 8. Portanto, o traço é 8.

Exemplo 2:

Vamos criar um tensor que tenha 2 linhas e 2 colunas na CPU e retorne o traço da matriz de tensor.

# Vamos importar módulo de tocha
importação de tocha
#Crie uma matriz de tensor 2D
Data1 = Torch.Tensor ([[2,33], [3,1]]).CPU()
#mostrar
Print ("Matriz Tensor Real:")
Imprimir (Data1)
Print ("Rastreio de uma matriz:")
#return rastreio
Imprimir (tocha.rastreamento (Data1))

Saída:

Tensor Matrix real:
Tensor ([[2, 33],
[3, 1]])
Traço de uma matriz:
Tensor (3)

A soma das diagonais é: 2+1 = 3. Portanto, o traço é 3.

Conclusão

Nesta lição de Pytorch, discutimos sobre a função Trace (). Retorna a soma dos elementos diagonais em uma matriz. Também discutimos os diferentes exemplos e trabalhamos esses exemplos em uma máquina de CPU.