Функция PyTorch torch.cumsum

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.cumsumявляется функцией в PyTorch, используемой для вычисления кумулятивной суммы. Она возвращает кумулятивную сумму вдоль указанного измерения, т. е. сумму всех элементов от начала до текущей позиции.

Определение функции

torch.cumsum(input, dim, dtype=None)

Пример использования

Пример

import torch

# Вычислить кумулятивную сумму
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumsum(x, dim=0)
print("Вход:", x)
print("Кумулятивная сумма:", result)
# Вывод: tensor([1, 3, 6, 10, 15])
# Пояснение: 1, 1+2=3, 1+2+3=6, 1+2+3+4=10, 1+2+3+4+5=15

# 2-мерный тензор
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Кумулятивная сумма по столбцам
result_col = torch.cumsum(x, dim=0)
print("\nКумулятивная сумма по столбцам:")
print(result_col)
# tensor([[ 1, 2, 3],
#         [ 5,  7,  9]])

# Кумулятивная сумма по строкам
result_row = torch.cumsum(x, dim=1)
print("\nКумулятивная сумма по строкам:")
print(result_row)
# tensor([[ 1, 3, 6],
#         [ 4,  9, 15]])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения