Pytorch 沿轴张量列表上的Pyrotch和

我有一个相同形状的张量列表。 我想把整个张量列表沿着一个轴求和。

torch.cumsum

是否沿dim执行此操作？

---

如果是这样，则需要将列表转换为单个张量并求和？

您不需要

cumsum

，

sum

是您的朋友 是的，你应该首先根据你的需要，用

stack

或

cat

将它们转换成一个张量，如下所示：

import torch
my_list = [torch.randn(3, 5), torch.randn(3, 5)]
result = torch.stack(my_list, dim=0).sum(dim=0).sum(dim=0)
print(result.shape) #torch.Size([5])

---

你的准确要求是什么？你想要什么样的输出？