阅读(16.9k) 书签 (0)

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