检查Tensor中是否包含值

Question

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

输入：

• test：目标
中可能存在的值的张量
• 目标：值的张量

输出：

• 输出：布尔的张量，与测试相同的形状。 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张量中成功找到的项目。