Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.keras.backend.cast

TensorFlow 1 version View source on GitHub

Casts a tensor to a different dtype and returns it.

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

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)