W3cubDocs

/TensorFlow 2.3

tf.keras.backend.cast

View source on GitHub

Casts a tensor to a different dtype and returns it.

You can cast a Keras variable but it still returns a Keras tensor.

Arguments
x Keras tensor (or variable).
dtype String, either ('float16', 'float32', or 'float64').
Returns
Keras tensor with dtype dtype.

Examples:

Cast a float32 variable to a float64 tensor

input = tf.keras.backend.ones(shape=(1,3))
print(input)
<tf.Variable 'Variable:0' shape=(1, 3) dtype=float32,
numpy=array([[1., 1., 1.]], dtype=float32)>
cast_input = tf.keras.backend.cast(input, dtype='float64')
print(cast_input)
tf.Tensor([[1. 1. 1.]], shape=(1, 3), dtype=float64)

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/keras/backend/cast