fix bug in ensure_one_hot
fix bug
fix bug
merge