Pytorch - Cumsum

Pytorch to open source framework dla języka programowania Pythona.

Tensor to wielowymiarowa tablica, która jest używana do przechowywania danych. Aby użyć tensora, musimy zaimportować moduł pochodni.

Aby utworzyć tensor, zastosowana metoda to TENSOR ().

Składnia:
latarka.tensor (dane)

Gdzie dane są wielowymiarową tablicą.

latarka.cumsum ()

latarka.cumsum () zwraca skumulowaną sumę elementów w dwuwymiarowym tensorze na rzędach lub w kolumnach.

Składnia:
latarka.cumsum (tensor_object, DIM)

Parametry:

1. Tensor_Object jako pierwszy parametr. Musi być dwuwymiarowy.
2. DIM = 0 Określa obliczenia kolumnowe i DIM = 1 Określa obliczenia rzędowe.

Przykład 1:
W tym przykładzie utworzymy tensor, który ma cztery wiersze i cztery kolumny i zwrócimy skumulowaną sumę każdego elementu w rzędu.

#Import Moduł pochodni
Importuj pochodnię
#Tensor
data1 = pochodnia.tensor ([[2,3,4,5], [1,3,5,3], [2,3,2,1], [2,3,4,2]]))
#wyświetlacz
Drukuj („Rzeczywisty tensor:”)
Drukuj (dane 1)
Drukuj („Sumulowana suma w całym rzędzie:”)
#Return Sumulacyjna suma
Drukuj (pochodnia.cumsum (dane 1,1))

Wyjście:

Rzeczywisty tensor:
tensor ([[2, 3, 4, 5],
[1, 3, 5, 3],
[2, 3, 2, 1],
[2, 3, 4, 2]])
Skumulowana suma w całym rzędzie:
tensor ([[2, 5, 9, 14],
[1, 4, 9, 12],
[2, 5, 7, 8],
[2, 5, 9, 11]))

Pracujący:
Rząd-1: 2,2+3,2+3+4,2+3+4+5 = [2, 5, 9, 14]
Row-2: 1,1+3,1+3+5,1+3+5+3 = [1, 4, 9, 12]

Row-3: 2,2+3,2+3+2,2+3+2+1 = [2, 5, 7, 8]
Row-4: 2,2+3,2+3+4,2+3+4+2 = [2, 5, 9, 11]

Przykład 2:
W tym przykładzie utworzymy tensor, który ma cztery wiersze i cztery kolumny i zwrócimy skumulowaną sumę każdego elementu w kolumnie.

#Import Moduł pochodni
Importuj pochodnię
#Tensor
data1 = pochodnia.tensor ([[2,3,4,5], [1,3,5,3], [2,3,2,1], [2,3,4,2]]))
#wyświetlacz
Drukuj („Rzeczywisty tensor:”)
Drukuj (dane 1)
Drukuj („skumulowana suma w całej kolumnie:”)
#Return Sumulacyjna suma
Drukuj (pochodnia.cumsum (dane 1,0))

Wyjście:

Rzeczywisty tensor:
tensor ([[2, 3, 4, 5],
[1, 3, 5, 3],
[2, 3, 2, 1],
[2, 3, 4, 2]])
Skumulowana suma w całej kolumnie:
tensor ([[2, 3, 4, 5],
[3, 6, 9, 8],
[5, 9, 11, 9],
[7, 12, 15, 11]))

Pracujący:
Kolumna 1: 2,2+1,2+1+2,2+1+2+2 = [2, 3, 5, 7]
Kolumna-2: 3,3+3,3+3+3,3+3+3+3 = [3,6,9,12]
Kolumna-3: 4,4+5,4+5+2,4+5+2+4 = [4,9,11,15]
Kolumna-4: 5,5+3,5+3+1,5+3+1+2 = [5,8,9,11]

Pracuj z CPU

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

W tej chwili, kiedy tworzymy tensor, możemy użyć funkcji CPU ().

Składnia:
latarka.tensor (dane).procesor()

Przykład 1:
W tym przykładzie utworzymy tensor na procesorze, który ma cztery wiersze i cztery kolumny i zwrócimy skumulowaną sumę każdego elementu w rzędzie.

#Import Moduł pochodni
Importuj pochodnię
#Tensor
data1 = pochodnia.tensor ([[2,3,4,5], [1,3,5,3], [2,3,2,1], [2,3,4,2]])).procesor()
#wyświetlacz
Drukuj („Rzeczywisty tensor:”)
Drukuj (dane 1)
Drukuj („Sumulowana suma w całym rzędzie:”)
#Return Sumulacyjna suma
Drukuj (pochodnia.cumsum (dane 1,1))

Wyjście:

Rzeczywisty tensor:
tensor ([[2, 3, 4, 5],
[1, 3, 5, 3],
[2, 3, 2, 1],
[2, 3, 4, 2]])
Skumulowana suma w całym rzędzie:
tensor ([[2, 5, 9, 14],
[1, 4, 9, 12],
[2, 5, 7, 8],
[2, 5, 9, 11]))

Pracujący:
Rząd-1: 2,2+3,2+3+4,2+3+4+5 = [2, 5, 9, 14]
Row-2: 1,1+3,1+3+5,1+3+5+3 = [1, 4, 9, 12]

Row-3: 2,2+3,2+3+2,2+3+2+1 = [2, 5, 7, 8]
Row-4: 2,2+3,2+3+4,2+3+4+2 = [2, 5, 9, 11]

Przykład 2:
W tym przykładzie utworzymy tensor na procesorze, który ma cztery wiersze i cztery kolumny i zwrócimy skumulowaną sumę każdego elementu w kolumnie.

#Import Moduł pochodni
Importuj pochodnię
#Tensor
data1 = pochodnia.tensor ([[2,3,4,5], [1,3,5,3], [2,3,2,1], [2,3,4,2]])).procesor()
#wyświetlacz
Drukuj („Rzeczywisty tensor:”)
Drukuj (dane 1)
Drukuj („skumulowana suma w całej kolumnie:”)
#Return Sumulacyjna suma
Drukuj (pochodnia.cumsum (dane 1,0))

Wyjście:

Rzeczywisty tensor:
tensor ([[2, 3, 4, 5],
[1, 3, 5, 3],
[2, 3, 2, 1],
[2, 3, 4, 2]])
Skumulowana suma w całej kolumnie:
tensor ([[2, 3, 4, 5],
[3, 6, 9, 8],
[5, 9, 11, 9],
[7, 12, 15, 11]))

Pracujący:
Kolumna 1: 2,2+1,2+1+2,2+1+2+2 = [2, 3, 5, 7]
Kolumna-2: 3,3+3,3+3+3,3+3+3+3 = [3,6,9,12]
Kolumna-3: 4,4+5,4+5+2,4+5+2+4 = [4,9,11,15]
Kolumna-4: 5,5+3,5+3+1,5+3+1+2 = [5,8,9,11]

Wniosek

W tym samouczku Pytorcha widzieliśmy, jak wykonać skumulowaną sumę na tensorze za pomocą pochodni.funkcja cumsum (). Zwraca skumulowaną sumę elementów w dwuwymiarowym tensorze w rzędach lub w kolumnach. Wdrożyliśmy również tę funkcję na procesorze za pomocą funkcji CPU ().