博客

tf.cast(x, dtype, name=None)

此函数是类型转换函数

参数:

• x：输入
• dtype：转换目标类型
• name：名称

返回：Tensor

例：

# tensor `a` is [1.8, 2.2], dtype=tf.float
tf.cast(a, tf.int32) ==> [1, 2]  # dtype=tf.int32

a = tf.Variable([1,0,0,1,1])
b = tf.cast(a,dtype=tf.bool)
sess = tf.Session()
sess.run(tf.initialize_all_variables())
print(sess.run(b))
#[ True False False  True  True]

参考：https://www.cnblogs.com/hezhiyao/p/8196587.html