tf.cast()

Ref:
https://blog.csdn.net/UESTC_C2_403/article/details/72190282

tf.cast用法

tf.cast：改變tensor的資料型態

例如：

import tensorflow as tf;








import numpy as np;








 








A = tf.convert_to_tensor(np.array([[1,1,2,4], [3,4,8,5]]))








 








with tf.Session() as sess:








 print A.dtype








 b = tf.cast(A, tf.float32)








 print b.dtype

输出：

<dtype: 'int64'>
<dtype: 'float32'>