tf.keras.backend.cast_to_floatx

View source on GitHub

Cast a Numpy array to the default Keras float type.

Arguments
x Numpy array.
Returns
The same Numpy array, cast to its new type.

Example:

from tensorflow.keras import backend as K
K.floatx()
    'float32'
arr = numpy.array([1.0, 2.0], dtype='float64')
arr.dtype
    dtype('float64')
new_arr = K.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/r1.15/api_docs/python/tf/keras/backend/cast_to_floatx