np.in1d的TensorFlow等效值

yzuktlbb 于 2023-03-13 发布在其他

关注(0)|答案(1)|浏览(138)

我正在努力做到：

a = [1,2,3,4,5,6]
b = [1,5]

result = [True,False,False,False,True,False]

也就是np.in1d函数https://docs.scipy.org/doc/numpy/reference/generated/numpy.in1d.html
有没有办法在TensorFlow中实现这一点？
谢谢！

tensorflow

来源：https://stackoverflow.com/questions/40945261/tensorflow-equivalent-of-np-in1d

1条答案

按热度按时间

djmepvbi1#

您可以将tf.equal与广播一起使用以形成5x2矩阵，其中i,j条目具有True，如果a[i]==b[j]，则tf.reduce_any折叠为布尔向量

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)

赞(0）回复(0）举报 2023-03-13

我来回答

np.in1d的TensorFlow等效值

1条答案

相关问题

热门标签

最新问答