torch.max(dim=?)
看shape=[2,3]的矩阵 在dim=0上取max其实得到了3个东西。所以应该是从dim=0看这三列每列的max。 类似于该维度坐标作为最内层循环取max的想法 直观的，该维度坐标作为x轴，其他维度作为遍历体，取出来该维度的shape reduce到1
Aquacolor
book
分类
keyboard_arrow_down
comment 评论区
star_outline 咱快来抢个沙发吧！