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