torch.max用法(指定维度dim) torch中的dim和numpy中的axis是一个东西,通过理解axis来理解dim,一举两得。 # pytorch 2023-10-01 269 点赞 4 评论 407 浏览