SoftMaxTree.gradInput is table (like input)
fixed memory leak