张量流量张量的元素比较

Question

我正在尝试做类似

的事情

a1=np.zeros((1,2))
a2=np.zeros((1,2))
a1==a2

输出：

array([[ True,  True]])

但有张量

注意：我正在为keras损失函数做这个，所以我无法将其转换为numpy数组并将其重新转换为张量 ，谢谢

Answer 1

您正在寻找＆＃34;元素明智的比较＆＃34;这里记录了：https://www.tensorflow.org/versions/r0.12/api_docs/python/control_flow_ops/comparison_operators

在你的情况下：

tf.equal(a1, a2)