TensorFlow函数教程：tf.keras.backend.cast

tf.keras.backend.cast函数

tf.keras.backend.cast(
    x,
    dtype
)

定义在：tensorflow/python/keras/backend.py。

将张量转换为不同的dtype并返回它。

您可以投射Keras变量，但它仍会返回Keras张量。

参数：

• x：Keras张量（或变量）。
• dtype：字符串，或者（'float16'，'float32'，或'float64'）。

返回：

dtype为`dtype`的Keras张量。