Как сравнить два массива, используя тензор потока? - PullRequest
1 голос
/ 31 мая 2019

Мне нужно сравнить два массива и получить либо 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)

печатает [False, False]

мое требование - получить значение true или false, а не массив.

Ответы [ 2 ]

2 голосов
/ 31 мая 2019

Я получил решение.

equality = tf.equal(X, y)   
reduce_t = tf.reduce_all(equality)
print(sess.run(reduce_t))
2 голосов
/ 31 мая 2019

Если вы хотите вернуть False, если любые ваших значений не равны, вы можете использовать операцию redu_all :

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