神经网络中_,predicted=中_,的作用

神经网络中_,predicted=中_,的作用

在神经网络训练中，经常会用到这样一段代码。

_,predicted=torch.max(output.data,dim=1)

其中_,是什么作用呢？
output是通过神经网络最后一层softmax函数作用之后的输出。例如，假设我们在分类问题中共有4类，分别是0，1，2，3。
假设output.data的第一行是[1,0,0,0]。（表明属于第一个类别0的概率是1)。
那么，假设不加_,代码

predicted=torch.max(output.data,dim=1)

输出的则是 1。（其中1是tensor)。加了 , 之后，输出的则变成了0。
因此，, 的作用是，是使predicted返回output.data行中最大数值所在位置代表的类别。如本例中代表所属类别0。