tf.cast()

时间:2022-06-19
本文章向大家介绍tf.cast()，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

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

随机文章
本站知识点必读
最近更新