seems incorrect, can not pass my test case@kazum ```python ####################################################################### # GatherNd # -------------------------- def _gather_nd(in_shape, indices): """test operator GatherNd""" np_indices = np.asarray(indices, dtype='int32') tf.reset_default_graph() with tf.Graph().as_default(): np_data = np.random.uniform(size=in_shape).astype("float32") in_data = tf.placeholder(tf.float32, in_shape, name="in_data") in_indices = tf.placeholder(tf.float32, np_indices.shape, name="in_indices") out = tf.gather_nd(in_data, indices) compare_tf_with_tvm([np_data, np_indices], ['in_data:0', 'in_indices:0'], out.name)
def test_forward_gather_nd(): _gather_nd((2, 3), [[0, 0], [1, 1]]) _gather_nd((2, 3), [[1], [0]]) _gather_nd((2, 3, 4), [[1]]) _gather_nd((4, 3, 2), [[0, 1], [1, 0]]) _gather_nd((2, 4), [[[0, 0]], [[0, 1]]]) _gather_nd((4, 2), [[[1]], [[0]]]) _gather_nd((2, 4, 2), [[[1]], [[0]]]) _gather_nd((2, 2, 3), [[[0, 1], [1, 0]], [[0, 0], [1, 1]]]) _gather_nd((2, 3, 3), [[[0, 0, 1], [1, 0, 1]], [[0, 1, 1], [1, 1, 0]]]) ``` --- [Visit Topic](https://discuss.tvm.ai/t/gather-nd-semantics/6243/5) 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/45fe43e3b19eb496a8603efd052477d9a746b011c3af10fcbaf4767e7e4b8fb2).