Pytorch 对沿轴的张量列表求和

【问题标题】：Pytorch sum over a list of tensors along an axisPytorch 对沿轴的张量列表求和
【发布时间】：2019-08-05 04:42:58
【问题描述】：

我有一个相同形状的张量列表。我想沿轴对整个张量列表求和。 torch.cumsum 是否在暗处执行此操作？如果是这样，它需要将列表转换为单个张量并求和？

【问题讨论】：

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

标签： pytorch

【解决方案1】：

你不需要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])

【讨论】：