Solved minor bugs of autodiff and dataset token