我遇到了一个看似微不足道的问题,但找不到解决方案。我有两个张量分别为thres和diff:
thres Tensor(“ truediv:0”,shape =(?,),dtype = float32)
diff Tensor(“ Sub:0”,shape =(?,),dtype = float32)
我想比较它们并返回numpy。我尝试将ind放入会话中并以numpy的形式返回。
但是我遇到了以下错误。
def f1():
ind = tf.constant(1)
return ind
def f2():
ind = tf.constant(0)
return ind
ind = tf.cond(tf.reduce_mean(thres - diff) < 0, f1, f2)
sess_1 = tf.compat.v1.Session()
print(sess_1.run(ind))
tensorflow.python.framework.errors_impl.InvalidArgumentError:您必须使用dtype uint8和shape输入占位符张量的值