tf.Assert() 根据条件打印数据
参数:
tf.Assert(
condition,//条件
data,//数据
summarize=None,
name=None
)
如果 condition 的结果为假,请打印 data 中的张量列表,summarize 用来确定要打印的张量的条目数量.
注意:为了确保断言执行,通常会附加依赖关系。
使用案例:
import tensorflow as tf
x = [1, 2, 3]
y = 1
assert_op = tf.Assert(tf.less_equal(tf.reduce_max(x), 3), x)//在x中的元素小于3时不报错并在下面输出1,在x大于3时抛出异常并打印x assertion failed: [1] [2] [4]
sess = tf.Session()
with tf.control_dependencies([assert_op]):
print(sess.run(tf.identity(1))