TensorFlow 張量類型轉(zhuǎn)換

tf.bitcast

bitcast ( 
    input, 
    type, 
    name = None
 )

參見指南：張量變換

在不復(fù)制數(shù)據(jù)的情況下,將張量從一種類型轉(zhuǎn)換到另一種類型.

給定張量輸入,此操作返回的張量與數(shù)據(jù)類型的輸入具有相同緩沖區(qū)信息.

如果輸入數(shù)據(jù)類型 T 大于輸出數(shù)據(jù)類型,則形狀將從 [...] 更改為 [...,sizeof(T)/ sizeof(type)].

如果 T 小于 type,則操作者要求最右邊的維度等于 sizeof(type)/ sizeof(T).然后形狀從 [...,sizeof(type)/ sizeof(T)] 轉(zhuǎn)到 [...].

注意：Bitcast 被當(dāng)為低級的計(jì)算,因此具有不同字節(jié)序的機(jī)器將給出不同的結(jié)果.

ARGS：

• input：張量.必須是下列類型之一：float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
• type：一個 tf.DType 來自：tf.float32, tf.float64, tf.int64, tf.int32, tf.uint8, tf.uint16, tf.int16, tf.int8, tf.complex64, tf.complex128, tf.qint8, tf.quint8, tf.qint32, tf.half.
• name：操作的名稱(可選).

返回：

返回 type 型張量.