这篇博客详细解释了PyTorch中torch.max函数的用法，特别是参数k和1的含义。当k不加下划线时，它返回矩阵每一行的最大值；而当k加下划线时，它会返回每一行中最大值对应的索引位置。通过实例演示，加深了对这一重要操作的理解。
torch.max(k,1)
其中这个 1代表行，0的话代表列。
不加_,返回的是一行中最大的数。
加_,则返回一行中最大数的位置。
测试如下：
_, predicted = torch.max(outputs, 1)，_,的作用
扫一扫
评论 1
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
