tf.cast

2018-04-09 本文已影响0人 Nise9s

tf.cast(
    x,
    dtype,
    name=None
)

tf.cast可以改变tensor的数据类型。
x是我们要操作的tensor，而dtype是我们要将tensor变成的数据类型。
举例如下：

x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, tf.int32)  # [1, 2], dtype=tf.int32

完成了将一个数据类型为float32得tensor变为int32
类型的tensor的操作。