Python Torch张量，第i个元素为所有前一个元素的乘积

我有一个像这样的火炬张量：

A = torch.tensor([1,2,3,4,5,6])

有一种简单的方法可以产生这样的张量吗

[1, 1*2, 1*2*3, 1*2*3*4, 1*2*3*4*5, 1*2*3*4*5*6]

非常感谢

您可以使用：

哪些产出：

tensor([1, 2, 6, 24, 120, 720])

---

你能发布函数cumprod（）的输出吗？

tensor([1, 2, 6, 24, 120, 720])