Created by: ForFishes

修复index的数据类型，有 int 修改为 Variable. issue: https://github.com/PaddlePaddle/Paddle/issues/21369