tf.cast()的用法(转)

2024年5月27日 626次阅读 来源: pytorch

一、函数 tf.cast()

cast(
    x,
    dtype,
    name=None
)

将x的数据格式转化成dtype.例如，原来x的数据格式是bool， 
那么将其转化成float以后，就能够将其转化成0和1的序列。反之也可以

 

二、例子

【code】

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]

————————————————————

参考：

  1. http://blog.csdn.net/luoganttcc/article/details/70315538
  2. https://www.tensorflow.org/api_docs/python/tf/cast
    原文作者：pytorch
    原文地址: https://www.cnblogs.com/hezhiyao/p/8196587.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞