tf.cast

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

参数:

• x: 张量或稀疏张量。
• dtype: 目标类型。
• name: 操作的名称(可选)。

返回值:

• 与x形状相同的张量或稀疏张量。

异常:

• TypeError: If x cannot be cast to the dtype.