dim的理解

eg.

a = torch.randn((3,4,5))
b = a.max(dim=0)

可以这么理解，固定这一个维度，比如现在是dim=0，这三个红框里面的部分不动，如红色箭头所示，移动整个块，进行比较或者其它操作。
输出维度：（1，4，5）

dim=1也是一样的

输出维度：（3，1，5）
最后的结果是将该维度变成1，其余不变。