Numpy 获取满足条件tensorflow的张量行

假设有上述代码。我想得到上面minibatch张量的行，这样每个检索到的行的第一个元素==a。如何在Tensorflow中执行此操作？另外，在Numpy中如何执行此操作？

（给定值a）

要在numpy中实现这一点，您只需编写：

input_mb = tf.placeholder(tf.int32, [None, 166, 1], name="input_minibatch")

在tensorflow中使用tf，其中：

selected_rows = myarray[myarray[:,0]== a]

---

我会在tensorflow上这样做：

mytensor[tf.squeeze(tf.where(tf.equal(mytensor[:,0],a), None, None))

tf.gather(mytensor, tf.squeeze(tf.where(tf.equal(mytensor[:,0],a), None, None)), axis=0)