python - 如何将 keras 网络更改为 channels_last 而不是 channels_first？

Question

更改网络是否期望通道是第一个还是最后一个的代码是什么？我知道你可以把它放在层本身，但我必须data_format='channels_last'每次都写吗？我多少记得有一种更简单的方法，但我忘记了，因此我的问题。

但是，这是一种方法：

model = tf.keras.Sequential([tf.keras.layers.Conv2D(1,(3,3),data_format="channels_last")])

标签： pythontensorflowkeras