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

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 是随机值的一维张量。

我正在尝试从训练集中随机抽取一批（实际上我正在尝试找出如何针对多个 epoch 进行训练，question here）并认为随机张量的索引是可行的方法，但是这不起作用。

Answer 1

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

答案似乎是 tf.gather 正是这样做的。