* Replace paddle.fluid.layers.gather_nd with paddle.gather_nd

* replace gather_nd with paddle.gather_nd

* fix the call of gather_nd

* fix code style of gather_nd