Pytorch - Trace

Wir werden sehen, wie wir in diesem Pytorch -Tutorial die Spur einer bestimmten Tenormatrix finden können.

Pytorch ist ein Open-Source-Framework mit einer Python-Programmiersprache. Tensor ist ein mehrdimensionales Array, mit dem die Daten gespeichert werden. Um einen Tensor zu verwenden, müssen wir das Fackelmodul importieren. Um einen Tensor zu erstellen, ist die verwendete Methode Tensor ().

Syntax:

Fackel.Tensor (Daten)

Wobei die Daten ein mehrdimensionales Array sind.

Fackel.verfolgen()

Trace wird als Summe der diagonalen Elemente in einer Matrix berechnet.

Syntax:

Fackel.Trace (Tensor_Object)

Parameter:

Es dauert ein Tensorobjekt als Parameter, bei dem es sich um einen 2D -Tensor handelt.

Beispiel 1:

Erstellen wir einen Tensor mit 4 Zeilen und 4 Spalten und geben die Spur der Tensormatrix zurück.

#Lassen Sie uns das Taschenmodul importieren
Taschenlampe importieren
#create eine 2D -Tensormatrix
Data1 = Fackel.Tensor ([[2,3,4,5] [3,1,2,3] [2,4,5,6] [5,6,7,0]])
#Anzeige
print ("tatsächliche Tensormatrix:")
Druck (Daten1)
print ("Trace einer Matrix:")
#Return Trace
Druck (Taschenlampe.Trace (Data1))

Ausgang:

Tatsächliche Tensormatrix:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Spur einer Matrix:
Tensor (8)

Die Summe der Diagonalen lautet: 2+1+5+0 = 8. Daher ist die Spur 8.

Beispiel 2:

Erstellen wir einen Tensor mit 2 Zeilen und 2 Spalten und geben die Spur der Tensormatrix zurück.

# Lassen Sie uns das Taschenmodul importieren
Taschenlampe importieren
#create eine 2D -Tensormatrix
Data1 = Fackel.Tensor ([[2,33], [3,1]])
#Anzeige
print ("tatsächliche Tensormatrix:")
Druck (Daten1)
print ("Trace einer Matrix:")
#Return Trace
Druck (Taschenlampe.Trace (Data1))

Ausgang:

Tatsächliche Tensormatrix:
Tensor ([[2, 33],
[3, 1]])
Spur einer Matrix:
Tensor (3)

Die Summe der Diagonalen lautet: 2+1 = 3. Daher ist die Spur 3.

Arbeiten Sie mit CPU

Wenn Sie die Trace () -Funktion auf der CPU ausführen möchten, müssen wir einen Tensor mit einer CPU () -Funktion erstellen. Dies wird auf einer CPU -Maschine ausgeführt.

Wenn wir einen Tensor erstellen, können wir diesmal die Funktion cpu () verwenden.

Syntax:

Fackel.Tensor (Daten).Zentralprozessor()

Beispiel 1:

Erstellen wir einen Tensor mit 4 Zeilen und 4 Spalten auf der CPU und geben die Spur der Tensormatrix zurück.

#Lassen Sie uns das Taschenmodul importieren
Taschenlampe importieren
#create eine 2D -Tensormatrix
Data1 = Fackel.Tensor ([[2,3,4,5] [3,1,2,3] [2,4,5,6] [5,6,7,0]]).Zentralprozessor()
#Anzeige
print ("tatsächliche Tensormatrix:")
Druck (Daten1)
print ("Trace einer Matrix:")
#Return Trace
Druck (Taschenlampe.Trace (Data1))

Ausgang:

Tatsächliche Tensormatrix:
Tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Spur einer Matrix:
Tensor (8)

Die Summe der Diagonalen lautet: 2+1+5+0 = 8. Daher ist die Spur 8.

Beispiel 2:

Erstellen wir einen Tensor mit 2 Zeilen und 2 Spalten auf der CPU und geben die Spur der Tensormatrix zurück.

# Lassen Sie uns das Taschenmodul importieren
Taschenlampe importieren
#create eine 2D -Tensormatrix
Data1 = Fackel.Tensor ([[2,33], [3,1]]).Zentralprozessor()
#Anzeige
print ("tatsächliche Tensormatrix:")
Druck (Daten1)
print ("Trace einer Matrix:")
#Return Trace
Druck (Taschenlampe.Trace (Data1))

Ausgang:

Tatsächliche Tensormatrix:
Tensor ([[2, 33],
[3, 1]])
Spur einer Matrix:
Tensor (3)

Die Summe der Diagonalen lautet: 2+1 = 3. Daher ist die Spur 3.

Abschluss

In dieser Pytorch -Lektion haben wir über die Trace () -Funktion besprochen. Es gibt die Summe der diagonalen Elemente in einer Matrix zurück. Wir haben auch die verschiedenen Beispiele besprochen und diese Beispiele auf einer CPU -Maschine bearbeitet.