tf.cast( )
cast直译,类似于映射,指映射到你制定的一个类型。
函数作用:
对输入tensor进行类型转换。语法:
tf.cast(x, dtype, name = None)
x:输入tensor
dtype:目标数据类型,将x的数据类型转化成dtype。
name:自定义输出tensor的名字。返回:
1个tensor例子:
x原来的数据类型是int,现在要把其转化成bool。
import tensorflow as tf
a = tf.Variable([1, 0, 0, 1, 1])
b = tf.cast(a, dtype = tf.bool)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(sess.run(b))
# 输出 [ True False False True True]