def unflatten(self, x): return special.from_onehot(x)
def unflatten(self, x): if config.TF_NN_SETTRACE: ipdb.set_trace() return special.from_onehot(x)
def unflatten(self, x): if np.isscalar(x): return x else: return special.from_onehot(x)