相关疑难解决方法(0)

在torch.sum() 中dim=-1 或-2 是什么意思?

让我以一个二维矩阵为例:

mat = torch.arange(9).view(3, -1)

tensor([[0, 1, 2],
        [3, 4, 5],
        [6, 7, 8]])

torch.sum(mat, dim=-2)

tensor([ 9, 12, 15])
Run Code Online (Sandbox Code Playgroud)

我发现的结果torch.sum(mat, dim=-2)等于torch.sum(mat, dim=0)dim=-1等于dim=1。我的问题是如何理解这里的负面维度。如果输入矩阵有 3 个或更多维度怎么办?

python pytorch

8
推荐指数
2
解决办法
6785
查看次数

标签 统计

python ×1

pytorch ×1