问题:
在tensorflow或者keras中使用==,例如
if Tensor_A == Tensor_B
时,会导致错误:
using a `tf.Tensor` as a Python `bool` is not allowed
解决方案:
这是因为==或!=等运算符返回的是bool Tensor,而不是python中普通的bool。可以使用is进行直接判断两个Tensor是否相同;或使用(Tensor_A==Tensor_B) is not None;如果要判断if a>b这样的,要使用tf.cond(a>b,lambda:process1,lambda:process2)