TensorFlow 1 version
|
View source on GitHub
|
Sets the value of the image data format convention.
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.
|