tf.keras.backend.cast_to_floatx
Cast a Numpy array to the default Keras float type.
tf.keras.backend.cast_to_floatx(
x
)
| Arguments | |
|---|---|
x |
Numpy array or TensorFlow tensor. |
| Returns | |
|---|---|
The same array (Numpy array if x was a Numpy array, or TensorFlow tensor if x was a tensor), cast to its new type. |
Example:
tf.keras.backend.floatx()
'float32'
arr = np.array([1.0, 2.0], dtype='float64')
arr.dtype
dtype('float64')
new_arr = cast_to_floatx(arr)
new_arr
array([1., 2.], dtype=float32)
new_arr.dtype
dtype('float32')
© 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_to_floatx