Pytorch - Trace

Vi vil se hvordan du finner spor etter en gitt tenormatrise i denne Pytorch -opplæringen.

Pytorch er et open source-ramme tilgjengelig med et Python-programmeringsspråk. Tensor er et flerdimensjonalt utvalg som brukes til å lagre dataene. For å bruke en tensor, må vi importere fakkelmodulen. For å lage en tensor er metoden som brukes tensor ().

Syntaks:

lommelykt.Tensor (data)

Der dataene er en flerdimensjonal matrise.

Lommelykt.spor ()

Spor beregnes som summen av diagonale elementer i en matrise.

Syntaks:

lommelykt.Trace (Tensor_Object)

Parameter:

Det tar et tensorobjekt som en parameter som er en 2D -tensor.

Eksempel 1:

La oss lage en tensor som har 4 rader og 4 kolonner og returnere spor av tensormatrisen.

#La oss importere fakkelmodulen
Importer fakkel
#Create en 2D -tensormatrise
data1 = fakkel.Tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]])
#vise
Print ("Faktisk tensormatrise:")
Print (Data1)
trykk ("Trace of a Matrix:")
#Return Trace
trykk (fakkel.Trace (Data1))

Produksjon:

Faktisk tensormatrise:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Spor av en matrise:
Tensor (8)

Summen av diagonaler er: 2+1+5+0 = 8. Derfor er sporet 8.

Eksempel 2:

La oss lage en tensor som har 2 rader og 2 kolonner og returnere spor av tensormatrisen.

# La oss importere fakkelmodulen
Importer fakkel
#Create en 2D -tensormatrise
data1 = fakkel.Tensor ([[2,33], [3,1]])
#vise
Print ("Faktisk tensormatrise:")
Print (Data1)
trykk ("Trace of a Matrix:")
#Return Trace
trykk (fakkel.Trace (Data1))

Produksjon:

Faktisk tensormatrise:
Tensor ([[2, 33],
[3, 1]])
Spor av en matrise:
Tensor (3)

Summen av diagonaler er: 2+1 = 3. Derfor er sporet 3.

Arbeid med CPU

Hvis du vil kjøre spor () -funksjonen på CPU, må vi lage en tensor med en CPU () -funksjon. Dette kjøres på en CPU -maskin.

Når vi lager en tensor, kan vi bruke CPU () -funksjonen denne gangen.

Syntaks:

lommelykt.Tensor (data).prosessor()

Eksempel 1:

La oss lage en tensor som har 4 rader og 4 kolonner på CPU -en og returnere spor av tensormatrisen.

#La oss importere fakkelmodulen
Importer fakkel
#Create en 2D -tensormatrise
data1 = fakkel.Tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]]).prosessor()
#vise
Print ("Faktisk tensormatrise:")
Print (Data1)
trykk ("Trace of a Matrix:")
#Return Trace
trykk (fakkel.Trace (Data1))

Produksjon:

Faktisk tensormatrise:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Spor av en matrise:
Tensor (8)

Summen av diagonaler er: 2+1+5+0 = 8. Derfor er sporet 8.

Eksempel 2:

La oss lage en tensor som har 2 rader og 2 kolonner på CPU og returnere spor av tensormatrisen.

# La oss importere fakkelmodulen
Importer fakkel
#Create en 2D -tensormatrise
data1 = fakkel.Tensor ([[2,33], [3,1]]).prosessor()
#vise
Print ("Faktisk tensormatrise:")
Print (Data1)
trykk ("Trace of a Matrix:")
#Return Trace
trykk (fakkel.Trace (Data1))

Produksjon:

Faktisk tensormatrise:
Tensor ([[2, 33],
[3, 1]])
Spor av en matrise:
Tensor (3)

Summen av diagonaler er: 2+1 = 3. Derfor er sporet 3.

Konklusjon

I denne Pytorch -leksjonen diskuterte vi om spor () -funksjonen. Det returnerer summen av diagonale elementer i en matrise. Vi diskuterte også de forskjellige eksemplene og arbeidet disse eksemplene på en CPU -maskin.