tf.cast：用于改變某個張量的數據類型

例如：

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'>

?

開始的時候定義A沒有給出類型，采用默認類型，整形。利用tf.cast函數就改為float類型
---------------------
作者：UESTC_C2_403
來源：CSDN
原文：https://blog.csdn.net/UESTC_C2_403/article/details/72190282
版權聲明：本文為博主原創文章，轉載請附上博文鏈接！

總結

以上是生活随笔為你收集整理的tf.cast用法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。