TensorFlow 张量类型转换
2018-09-05 18:18 更新
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 型张量.