如何使用 tensorflow 比较两个数组?

问题描述

I need to compare two arrays and get either true or false,not elementwise result. My code is

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)

prints [False, False]

my requirement is to get true or false, not an array.

解决方案

Assuming that you want to return False if any of your values are not equal then you can use the reduce_all operation:

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