通过张量随机索引数组选择或切片张量？

Question

我试图在TensorFlow中复制它：

mat = np.matrix("1,2,3;4,5,6;7,8,9")
matrix([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])
mat[2]
matrix([[7, 8, 9]])

我收到此错误：

X_batch = X_train[batch]

     

IndexError：只有整数，切片（:），省略号（...），numpy.newaxis（None）和整数或布尔数组才是有效索引

X_train为batch_size x (other dimensions)，batch为随机值的1-d张量。

我试图从训练集中取一个随机批次（我真的想弄清楚如何训练多个时代，question here）并认为指数的随机张量是要走的路，但这不起作用。

Answer 1

我想我找到了这个类似问题的答案：Tensorflow: Using tf.slice to split the input

答案似乎是tf.gather，这正是如此。