fixed SoftMaxTree:updateOutput