[quote="adobay, post:5, topic:6243"] ``` in_indices = tf.placeholder(tf.float32, np_indices.shape, name="in_indices") out = tf.gather_nd(in_data, indices) ``` [/quote]
These lines should be ``` in_indices = tf.placeholder(tf.int32, np_indices.shape, name="in_indices") out = tf.gather_nd(in_data, in_indices) ``` then, your test worked correctly on my environment. --- [Visit Topic](https://discuss.tvm.ai/t/gather-nd-semantics/6243/6) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/4472b194ede7fe476088acd69a66c79b97a5fd8b6652494f84fc4610ddcc131c).