working on softmaxtree that does not use SparseOutLinear