Created by: typhoonzero

Fix https://github.com/PaddlePaddle/Paddle/issues/3759

Still got problems when using a 3d tensor input.