- if #self.outputModules == 1 then
- self.output = self.outputModules[1].output
- else
- self.output = { }
- for i, d in ipairs(self.outputModules) do
- self.output[i] = d.output
+ self.output = self:applyOnModules(function(m) return m.output end, self.outputModules)
+
+ return self.output
+end
+
+function DAG:updateGradInput(input, gradOutput)
+ self:sort()
+
+ self:applyOnModules(function(m, i, go) m:updateGradInput(i, go) end, self.outputModules, input, gradOutput)
+
+ for k = self.sorted, 1, -1 do
+ local m = sorted[k]
+ if self.succ[d] then
+ if #self.succ[d] == 1 then
+ d:updateGradInput(self.succ[d][1].gradInput)
+ elseif #self.succ[d] > 1 then
+ local sum
+ for k = 1, #self.succ[d] do
+ if sum then
+ sum:add(self.succ[d][k].gradInput)
+ else
+ sum = self.succ[d][k].gradInput:clone()
+ end
+ end
+ d:updateGradInput(sum)
+ end