tf.cast(
x,
dtype,
name=None
)
将张量投射到一个新的类型上。这个操作对x(对于张量)或x进行了强制转换。值(在sparse张量的情况下)到dtype。
例:
x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, tf.int32) # [1, 2], dtype=tf.int32
参数:
返回值:
异常:
TypeError
: If x
cannot be cast to the dtype
.