local CriterionTable, parent = torch.class('nn.CriterionTable', 'nn.Module')

function CriterionTable:__init(criterion)

parent.__init(self)

self.criterion = criterion

self.gradInput = {criterion.gradInput}

end

function CriterionTable:updateOutput(input)

self.output = self.criterion:updateOutput(table.unpack(input))

return self.output

end

function CriterionTable:updateGradInput(input, gradOutput)

self.criterion:updateGradInput(table.unpack(input))

return self.gradInput

end