tf.cast( )

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]
 

猜你喜欢

转载自blog.csdn.net/weixin_38145317/article/details/88714575
tf