tensorflow_api_1:tf.cast( )

来源:互联网 发布:ubuntu gcc离线安装包 编辑:程序博客网 时间:2024/06/03 19:16

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]
原创粉丝点击