TensorFlow 张量类型转换

tf.bitcast

bitcast ( 
    input, 
    type, 
    name = None
 )

参见指南：张量变换

在不复制数据的情况下,将张量从一种类型转换到另一种类型.

给定张量输入,此操作返回的张量与数据类型的输入具有相同缓冲区信息.

如果输入数据类型 T 大于输出数据类型,则形状将从 [...] 更改为 [...,sizeof(T)/ sizeof(type)].

如果 T 小于 type,则操作者要求最右边的维度等于 sizeof(type)/ sizeof(T).然后形状从 [...,sizeof(type)/ sizeof(T)] 转到 [...].

注意：Bitcast 被当为低级的计算,因此具有不同字节序的机器将给出不同的结果.

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 型张量.