SoftMaxTree.gradInput is a table (like input)