How to do the tf.gather_nd in pytorch?

[erwin (Erwin Wang)]

hello , do you solve it? I also meets the problem