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

由 Carrie 创建, 最后一次修改 2019-03-05

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张量。

以上内容是否对您有帮助:
二维码
建议反馈
二维码