torch.sum()维度0，1，2。比如现在有\(3\times\ 2\times3\)的张量，理解为3个\(2\times3\)的矩阵。当dim=0，1，2时分别在哪个维度上相加[1]？下面是具体的矩阵
\[[1,2,3]\\ [4,5,6]\\\\
[1,2,3] \\ [4,5,6]\\\\
[1,2,3] \\ [4,5,6]
\]
在哪个维度相加，那个维度就去掉。\(3\times2\times3\)分别就对应0，1，2三个维度。
- dim=0，最后计算结果就是\(2\times3\)。（可视化后按照宽维度相加对应元素）
- dim=1，最后计算结果就是\(3\times3\)。（可视化后按照高维度相加对应元素）
- dim=2，最后计算结果就是\(3\times2\)。（可视化后按照长维度相加对应元素）
宽和高维度是正面看的，所以不用动。而长维度是横着看，所以最后元素需要向左旋转。（具体计算时理解的，我这么表述可能不清楚）
示例代码
import torch
c = torch.tensor([[[1,2,3],
[4,5,6]],
[[1,2,3],
[4,5,6]],
[[1,2,3],
[4,5,6]]])
print(f" c size = {c.size()}")
c1=torch.sum(c , dim=0)
print(f" c1 = {c1}\n c1 size = {c1.size()}")
c2=torch.sum(c , dim=1)
print(f" c2 = {c2}\n c2 size = {c2.size()}")
c3=torch.sum(c , dim=2)
print(f" c3 = {c3}\n c3 size = {c3.size()}")
运行结果如下
