我希望检查一组给定的值是否包含在一个稀疏张量中。该稀疏张量称为标签
,只有一个维度,包含一个id列表。
最终,这似乎是一个简单的集合交集问题,因此我尝试了以下解决方法。
sparse_ids = load_ids_as_sparse_tensor()
wanted_ids = tf.constant([34, 56, 12])
intersection = tf.sets.set_intersection(
wanted_ids,
tf.cast(sparse_ids.values, tf.int32)
)
contains_any_wanted_ids = tf.not_equal(tf.size(intersection), 0)
然而,我遇到了这个错误:
ValueError: Shape must be at least rank 2 but is rank 1 for 'DenseToDenseSetOperation' (op: 'DenseToDenseSetOperation') with input shapes: [3], [?].
你有什么想法吗?