确定A值是否在TensorFlow中的Set中

的tf.logical_or,tf.logical_and和tf.select功能是非常有用的.

但是,假设你有价值x,并且你想看看它是否在set(a, b, c, d, e).在python中你只需写:

if x in set([a, b, c, d, e]):
  # Do some action.

据我所知,在TensorFlow中执行此操作的唯一方法是嵌套'tf.logical_or'和'tf.equal'.我在下面提供了这个概念的一次迭代:

tf.logical_or(
    tf.logical_or(tf.equal(x, a), tf.equal(x, b)),
    tf.logical_or(tf.equal(x, c), tf.equal(x, d))
)

我觉得在TensorFlow中必须有一种更简单的方法.在那儿？