使用keras时input_shape的维度表示问题说明
Keras提供了两套后端，Theano和Tensorflow，不同的后端使用时维度顺序dim_ordering会有冲突。
对于一张224*224的彩色图片表示问题，theano使用的是th格式，维度顺序是(3，224，224)，即通道维度在前，Caffe采取的也是这种方式。而Tensorflow使用的是tf格式，维度顺序是(224，224，3)，即通道维度在后。
Keras默认使用的是Tensorflow。我们在导入模块的时候可以进行查看，也可以切换后端。
为了代码可以在两种后端兼容，可以通过data_format参数进行维度顺序的设定，data_format='channels_first'，对应“th”，data_format='channels_last'，对应“tf”。
补充知识：Tensorflow Keras 中input_shape引发的维度顺序冲突问题（NCHW与NHWC)
以tf.keras.Sequential构建卷积层为例：
tf.keras.layers.Conv2D(10, 3, input_shape=(2, 9, 9),padding='same',activation=tf.nn.relu,kernel_initializer='glorot_normal', bias_initializer='glorot_normal'),
这是一个简单的卷积层的定义，主要看input_shape参数：
这是用来指定卷积层输入形状的参数，由于Keras提供了两套后端，Theano和Tensorflow，不同的后端使用时对该参数所指代的维度顺序dim_ordering会有冲突。
Theano(th)：
NCHW：顺序是 [batch, in_channels, in_height, in_width]
Tensorflow(tf)：keras默认使用这种方式
NHWC：顺序是 [batch, in_height, in_width, in_channels]
即对于上述input_shape=(2, 9, 9)来说：我们先忽略batch，2会被解析为通道数，矩阵大小为9*9，符合我们预期。而tf会将矩阵大小解析为2 * 9 ，且最后一位9代表通道数，与预期不符。
解决
法一：
在卷积层定义中加入参数来让keras在两种后端之间切换：
data_format='channels_first'：代表th
data_format='channels_last'：代表tf
但是该法在某些时候不成功会报错：
或许是cpu电脑导致的，只支持NHWC即tf模式。
只能修改相应文件的配置来使其支持NCHW，参考这里
法二：(推荐)
使用tf.transpose函数进行高维数据的转置（维度大于2，轴的转换）
如将上述(2,9,9)转为(9,9,2)并且是以2为通道数，即矩阵为9*9，而不是像reshape函数简单的调整维度，若使用reshape函数来转换，只会得到通道数为9，矩阵为9 * 2的数据。
tf.transpose(待转矩阵,(1,2,0))
解释：
其中0，1，2…是原矩阵维度从左到右轴的标号，即（2，9，9）中三个维度分别对应标号0，1，2。而调整过后将标号顺序变为1，2，0 即是把表通道数的轴置于最后，这样转置后的矩阵就满足了keras的默认tf后端。即可正常训练。
以上这篇使用keras时input_shape的维度表示问题说明就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持软件开发网。
您可能感兴趣的文章:keras获得model中某一层的某一个Tensor的输出维度教程在keras 中获取张量 tensor 的维度大小实例keras 获取某层的输入/输出 tensor 尺寸操作keras.layer.input()用法说明Tensorflow与Keras自适应使用显存方式
