Created by: guoshengCS
Fix the data lod in Transformer prediction, since lod_reset would check the input lod.