View source on GitHub |
Cast a Numpy array to the default Keras float type.
tf.keras.backend.cast_to_floatx( x )
Arguments | |
---|---|
x | Numpy array. |
Returns | |
---|---|
The same Numpy array, cast to its new type. |
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