Python tf.cast用法及代碼示例

將張量轉換為新類型。

用法

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

該操作將 x(在 Tensor 的情況下)或 x.values(在 SparseTensor 或 IndexedSlices 的情況下)轉換為 dtype 。

例如：

x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, tf.int32)
<tf.Tensor:shape=(2,), dtype=int32, numpy=array([1, 2], dtype=int32)>

注意 tf.cast 有一個別名 tf.dtypes.cast ：

x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.dtypes.cast(x, tf.int32)
<tf.Tensor:shape=(2,), dtype=int32, numpy=array([1, 2], dtype=int32)>

該操作支持數據類型(對於x和dtype) 的uint8,uint16,uint32,uint64,int8,int16,int32,int64,float16,float32,float64,complex64,complex128,bfloat16.如果從複雜類型轉換(complex64,complex128) 到實數類型，隻有實數部分x被退回。如果從真實類型轉換為複雜類型(complex64,complex128)，返回值的虛部設置為0.這裏對複雜類型的處理與 numpy 的行為相匹配。

請注意，將 nan 和 inf values 轉換為整數類型具有未定義的行為。