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 转换为整数类型具有未定义的行为。