神经网络中_,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。