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