What does tf.gather_nd intuitively do?
Ok, so think about it like this: You are providing a list of index values to index the provided tensor to get those slices. The first dimension of the indices you provide is for each index you will perform. Let’s pretend that tensor is just a list of lists. [[0]] means you want to get … Read more