小编And*_*rew的帖子

Pytorch 对沿轴的张量列表求和

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

pytorch

5
推荐指数
1
解决办法
1万
查看次数

标签 统计

pytorch ×1