eg.
a = torch.randn((3,4,5))
b = a.max(dim=0)
可以这么理解,固定这一个维度,比如现在是dim=0,这三个红框里面的部分不动,如红色箭头所示,移动整个块,进行比较或者其它操作。
输出维度:(1,4,5)
dim=1也是一样的
输出维度:(3,1,5)
最后的结果是将该维度变成1,其余不变。
eg.
a = torch.randn((3,4,5))
b = a.max(dim=0)
可以这么理解,固定这一个维度,比如现在是dim=0,这三个红框里面的部分不动,如红色箭头所示,移动整个块,进行比较或者其它操作。
输出维度:(1,4,5)
dim=1也是一样的
输出维度:(3,1,5)
最后的结果是将该维度变成1,其余不变。