PyTorch torch.cumsum-Funktion
PyTorch torch Referenzhandbuch
torch.cumsumist eine Funktion in PyTorch, die zur Berechnung der kumulativen Summe verwendet wird. Sie gibt die kumulative Summe entlang einer angegebenen Dimension zurück, d. h. die Summe aller Elemente vom Anfang bis zur aktuellen Position.
Funktionsdefinition
torch.cumsum(input, dim, dtype=None)
Anwendungsbeispiel
Beispiel
import torch
# Berechne die kumulative Summe
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumsum(x, dim=0)
print("Eingabe:", x)
print("Kumulative Summe:", result)
# Ausgabe: tensor([1, 3, 6, 10, 15])
# Erklärung: 1, 1+2=3, 1+2+3=6, 1+2+3+4=10, 1+2+3+4+5=15
# 2-dimensionaler Tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Kumulativ entlang der Spalten
result_col = torch.cumsum(x, dim=0)
print("\nKumulative Summe entlang Spalten:")
print(result_col)
# tensor([[ 1, 2, 3],
# [ 5, 7, 9]])
# Kumulativ entlang Zeilen
result_row = torch.cumsum(x, dim=1)
print("\nKumulative Summe entlang Zeilen:")
print(result_row)
# tensor([[ 1, 3, 6],
# [ 4, 9, 15]])
# Berechne die kumulative Summe
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumsum(x, dim=0)
print("Eingabe:", x)
print("Kumulative Summe:", result)
# Ausgabe: tensor([1, 3, 6, 10, 15])
# Erklärung: 1, 1+2=3, 1+2+3=6, 1+2+3+4=10, 1+2+3+4+5=15
# 2-dimensionaler Tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Kumulativ entlang der Spalten
result_col = torch.cumsum(x, dim=0)
print("\nKumulative Summe entlang Spalten:")
print(result_col)
# tensor([[ 1, 2, 3],
# [ 5, 7, 9]])
# Kumulativ entlang Zeilen
result_row = torch.cumsum(x, dim=1)
print("\nKumulative Summe entlang Zeilen:")
print(result_row)
# tensor([[ 1, 3, 6],
# [ 4, 9, 15]])
Andere Erweiterungen