tf.assert_same_type()判断x是否是tf_type类型的值
参数:
tf.assert_type(
tensor,
tf_type,
message=None,
name=None
)
使用案例:
import tensorflow as tf
x = 1.0
sess = tf.Session()
with tf.control_dependencies([tf.assert_type(x, tf.float32)]):
output = tf.reduce_sum(x)
print(sess.run(output))