嗨,我是使用Torch' nn'包。在过去的两周里,我对Torch' nn'中的gradInput和gradOutput的含义感到非常困惑。图书馆。我相信毕业生'这里的意思是梯度,但这两个变量究竟是指什么?
感谢任何人的帮助!
答案 0 :(得分:2)
gradOutput:渐变w.r.t.模块的输出。这可以从损失函数传递,也可以从当前模块旁边的模块传递。它用于计算梯度w.r.t.输入(gradInput)和渐变w.r.t.模块的参数(gradWeight / gradBias)