W3cubDocs

/TensorFlow Python

tf.keras.backend.set_floatx

tf.keras.backend.set_floatx(value)

Defined in tensorflow/python/keras/_impl/keras/backend.py.

Sets the default float type.

Arguments:

  • value: String; 'float16', 'float32', or 'float64'.

Example:

>>> from keras import backend as K
>>> K.floatx()
'float32'
>>> K.set_floatx('float16')
>>> K.floatx()
'float16'

Raises:

  • ValueError: In case of invalid value.

© 2018 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/api_docs/python/tf/keras/backend/set_floatx