TensorFlow等效的np.in1d

Question

我試圖做：

一個= [1,2,3,4,5,6] B = [1,5]

結果= [真，假，假，假，真，假]

這是np.in1d功能https://docs.scipy.org/doc/numpy/reference/generated/numpy.in1d.html

是否有TensorFlow實現這個的方法嗎？

謝謝！

Answer 1

您可以使用tf.equal與廣電形成5x2矩陣，其中i,j項具有True如果a[i]==b[j]，然後tf.reduce_any崩潰爲bool向量

a = [1,2,3,4,5,6] 
b = [1,5] 
a0 = tf.expand_dims(a, 1) 
b0 = tf.expand_dims(b, 0) 
result = sess.run(tf.reduce_any(tf.equal(a0, b0), 1)) 
assert result == np.in1d(a, b)