Функция PyTorch torch.cumprod

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

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

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

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

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

Пример

import torch

# Вычислить кумулятивное произведение
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumprod(x, dim=0)
print("Вход:", x)
print("Кумулятивное произведение:", result)
# Вывод: tensor([1, 2, 6, 24, 120])
# Объяснение: 1, 1*2=2, 1*2*3=6, 1*2*3*4=24, 1*2*3*4*5=120

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

# Кумулятивно по строке
result_row = torch.cumprod(x, dim=1)
print("nКумулятивное произведение по строке:")
print(result_row)
# tensor([[ 1, 2, 6],
#         [  4,  20, 120]])

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

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