python - 如何使用 tensorflow 比较两个数组？

Question

我需要比较两个数组并得到 true 或 false，而不是元素结果。我的代码是

X = tf.constant([0.05, 0.10], dtype=tf.float32, shape=[1, 2])
y = tf.constant([0.01, 0.99], dtype=tf.float32, shape=[1, 2])

equality = tf.equal(X, y)

打印[假，假]

我的要求是获取 true 或 false，而不是数组。

Answer 1

假设您希望在任何值不相等时返回False，那么您可以使用reduce_all操作:

equality = tf.math.reduce_all(tf.equal(X, y))