tf.keras.backend.set_floatx

tf.keras.backend.set_floatx(value)

Defined in tensorflow/python/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.