检查Tensor中是否包含值

Question

不幸的是我无法找到实现以下功能的功能:

输入:

• test:目标中可能存在的值的张量
• target:Tensor of values

输出:

• 输出:布尔的张量,与测试相同的形状. output[i] = targets.contains(test[i])

---

说清楚:我需要测试Tensor的元素是否包含在另一个Tensor中.

Answer 1

你想用tf.listdiff.

给定两个张量test和target,

c, _ = tf.listdiff(test, target)

c将包含测试中不在目标中的项目.然后你可以再次列出它以找到那里的那些:

output, _ = tf.listdiff(test, c)

output包含test在target张量中成功找到的项目.