tf.cast(x, dtype, name=None)

引數

• x：輸入
• dtype：轉換目標型別
• name：名稱

返回：Tensor

例子：

import tensorflow as tf

a = [1,0,1,0]
b = [1,2,3,4]
c = [True, True, False]
d = tf.cast(a, dtype=bool)
e = tf.cast(b, dtype=bool)
f = tf.cast(c, dtype=tf.float32)
sess = tf.InteractiveSession()
print(sess.run(d))  # [ True False  True False]
print(sess.run(e))  # [ True  True  True  True]
print(sess.run(f))  # [1. 1. 0.]