Pytorch - śledzenie

Zobaczymy, jak znaleźć ślad danej matrycy tenorowej w tym samouczku Pytorcha.

Pytorch to open source dostępny z językiem programowania Python. TENSOR to wielowymiarowa tablica używana do przechowywania danych. Aby użyć tensor, musimy zaimportować moduł pochodni. Aby utworzyć tensor, zastosowaną metodą jest tensor ().

Składnia:

latarka.tensor (dane)

Gdzie dane są wielowymiarową tablicą.

Latarka.namierzać()

Ślad jest obliczany jako suma pierwiastków przekątnych w matrycy.

Składnia:

latarka.Trace (Tensor_Object)

Parametr:

Zajmuje obiekt tensorowy jako parametr, który jest tensor 2D.

Przykład 1:

Utwórzmy tensor, który ma 4 wiersze i 4 kolumny i zwróć ślad macierzy tensorowej.

#Zaimportujmy moduł pochodni
Importuj pochodnię
#Utwórz macierz tensor 2D
data1 = pochodnia.tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]]))
#wyświetlacz
Drukuj („Rzeczywista macierz tensorowa:”)
Drukuj (dane 1)
Drukuj („Trace of a Matrix:”)
#return Trace
Drukuj (pochodnia.Trace (data1))

Wyjście:

Rzeczywista macierz tensorowy:
tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Ślad macierzy:
tensor (8)

Suma przekątnej wynosi: 2+1+5+0 = 8. Stąd ślad wynosi 8.

Przykład 2:

Utwórzmy tensor, który ma 2 wiersze i 2 kolumny i zwróć ślad macierzy tensor.

# Zaimportujmy moduł pochodni
Importuj pochodnię
#Utwórz macierz tensor 2D
data1 = pochodnia.tensor ([[2,33], [3,1]])
#wyświetlacz
Drukuj („Rzeczywista macierz tensorowa:”)
Drukuj (dane 1)
Drukuj („Trace of a Matrix:”)
#return Trace
Drukuj (pochodnia.Trace (data1))

Wyjście:

Rzeczywista macierz tensorowy:
tensor ([[2, 33],
[3, 1]])
Ślad macierzy:
tensor (3)

Suma przekątnej wynosi: 2+1 = 3. Stąd ślad wynosi 3.

Pracuj z CPU

Jeśli chcesz uruchomić funkcję trace () na procesorze, musimy utworzyć tensor z funkcją cpu (). Będzie to działać na komputerze procesora.

Kiedy tworzymy tensor, tym razem możemy użyć funkcji CPU ().

Składnia:

latarka.tensor (dane).procesor()

Przykład 1:

Utwórzmy tensor, który ma 4 wiersze i 4 kolumny na procesorze i zwróć ślad macierzy tensorowej.

#Zaimportujmy moduł pochodni
Importuj pochodnię
#Utwórz macierz tensor 2D
data1 = pochodnia.tensor ([[2,3,4,5], [3,1,2,3], [2,4,5,6], [5,6,7,0]])).procesor()
#wyświetlacz
Drukuj („Rzeczywista macierz tensorowa:”)
Drukuj (dane 1)
Drukuj („Trace of a Matrix:”)
#return Trace
Drukuj (pochodnia.Trace (data1))

Wyjście:

Rzeczywista macierz tensorowy:
tensor ([[2, 3, 4, 5],
[3, 1, 2, 3],
[2, 4, 5, 6],
[5, 6, 7, 0]])
Ślad macierzy:
tensor (8)

Suma przekątnej wynosi: 2+1+5+0 = 8. Stąd ślad wynosi 8.

Przykład 2:

Utwórzmy tensor, który ma 2 wiersze i 2 kolumny na procesorze i zwróć ślad macierzy tensor.

# Zaimportujmy moduł pochodni
Importuj pochodnię
#Utwórz macierz tensor 2D
data1 = pochodnia.tensor ([[2,33], [3,1]]).procesor()
#wyświetlacz
Drukuj („Rzeczywista macierz tensorowa:”)
Drukuj (dane 1)
Drukuj („Trace of a Matrix:”)
#return Trace
Drukuj (pochodnia.Trace (data1))

Wyjście:

Rzeczywista macierz tensorowy:
tensor ([[2, 33],
[3, 1]])
Ślad macierzy:
tensor (3)

Suma przekątnej wynosi: 2+1 = 3. Stąd ślad wynosi 3.

Wniosek

W tej lekcji Pytorcha omówiliśmy funkcję trace (). Zwraca sumę elementów przekątnych w matrycy. Omówiliśmy również różne przykłady i opracowaliśmy te przykłady na komputerze procesora.