tensorflow官方文档解释如下:
tf.cast(
x,
dtype,
name=None
)
tf.cast(
x,
dtype,
name=None
)
Casts a tensor to a new type.
The operation casts x
(in case of Tensor
) or x.values
(in case of SparseTensor
) to dtype
.
For example:
x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, tf.int32) # [1, 2], dtype=tf.int32