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