在pytest上运行时,tensorflow运行函数不返回numpy

时间:2018-09-19 16:38:23

标签: numpy tensorflow batch-normalization

使用苗条的tensorflow运行batch_norm并尝试打印numpy结果时得到下一个错误:

TypeError: '<' not supported between instances of 'dict' and 'int'

代码是:

x = tf.Variable(tf.truncated_normal(shape=(1, 2, 2, 32), dtype=tf.float32))
y = slim.batch_norm(x)
sess = tf.Session()
tf.global_variables_initializer().run(session=sess)
y_np = y.eval(session=sess)
x_np = x.eval(session=sess)
print("x\n" + str(x_np))
print("y\n" + str(y_np))

我也尝试过

sess = tf.Session()
tf.global_variables_initializer().run(session=sess)
y_np, x_np = sess.run([y, x])
print("x\n" + str(x_np))
print("y\n" + str(y_np))

有什么建议吗?

更新:

事实证明,仅当在pytest中运行时,if才能打印(我这样做了)。仅通过python运行时,打印成功。

0 个答案:

没有答案