tf.keras.backend.set_image_data_format

TensorFlow 1 version

View source on GitHub

Sets the value of the image data format convention.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.backend.set_image_data_format

tf.keras.backend.set_image_data_format(
    data_format
)

Args
data_format	string. 'channels_first' or 'channels_last'.

Example:

tf.keras.backend.image_data_format()
'channels_last'
tf.keras.backend.set_image_data_format('channels_first')
tf.keras.backend.image_data_format()
'channels_first'
tf.keras.backend.set_image_data_format('channels_last')

Raises
ValueError	In case of invalid data_format value.