高端索引tf.gather_nd(params, indices)

import tensorflow as tf
import numpy as np

params = np.arange(23)
a = tf.gather(params,1)

with tf.Session() as sess:
    print(sess.run(a))
   
params = np.arange(24).reshape(2,-1)
a = tf.gather(params,1)

with tf.Session() as sess:
    print(sess.run(a))

b = tf.gather_nd(params,[1,3])
with tf.Session() as sess:
    print(sess.run(b))

1
[12 13 14 15 16 17 18 19 20 21 22 23]
15

更多数学原理小文请关注公众号：未名方略