arr = tf.range(60) t = tf.reshape(arr, [3, 4, 5]) t1 = tf.gather_nd(t, [0]) t2 = tf.gather_nd(t, [0, 1]) t3 = tf.gather_nd(t, [0, 1, 2]) t4 = tf.gather_nd(t, [[0, 1, 2]]) t5 = tf.gather_nd(t, [[0, 0], [1, 1]]) t6 = tf.gather_nd(t, [[0, 0], [1, 1], [2, 2]]) t7 = tf.gather_nd(t, [[0, 0, 0], [1, 1, 1], [2, 2, 2]]) t8 = tf.gather_nd(t, [[[0, 0, 0], [1, 1, 1], [2, 2, 2]]]) print(t) print(t1) print(t2) print(t3) print(t4) print(t5) print(t6) print(t7) print(t8)
|
近期评论