Tensorflow学习之tf.cast类型转换函数

最新推荐文章于 2025-06-18 16:58:08 发布

序木

最新推荐文章于 2025-06-18 16:58:08 发布

阅读量9.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： Tensorflow学习 tf.cast类型转换函数

本文链接：https://blog.youkuaiyun.com/cc1949/article/details/78426206

深度学习专栏收录该内容

19 篇文章

订阅专栏

本文介绍TensorFlow中tf.cast函数的使用方法，该函数用于在TensorFlow程序中转换张量的数据类型。文章通过一个实例展示了如何将整数类型的张量转换为布尔类型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.cast(x, dtype, name=None) //此函数是类型转换函数，把x的类型转换成dtype指定的类型

Args:

x: A Tensor or SparseTensor.
dtype: The destination type.
name: A name for the operation (optional).
Returns:

A Tensor or SparseTensor with same shape as x.

Raises:

TypeError: If x cannot be cast to the dtype.

参数

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

Tensor

例子

#将0/1转换成bool类型

import tensorflow as tf

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]