Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Gradient for gather_nd is not implemented #5342
when I use
I can temporarily solve this problem to flatten both of
Is there any plan to support that? Thanks!
@tfka Yes, to use tf.gather you just need to reshape your params and indices.
gather_nd gradients are available on master, and should be in the latest minor release: