Conv2D中Keras默认input格式和Pytorch input格式的差异
- Keras的Conv2D的默认输入格式是 batch, width, height, channel
- Pytorch的Conv2D是 batch, channel, width, height
下面是图例
keras:
keras官方文档的输入格式,注意看到data_format这个属性,修改data_format为"channels_first"和"channels_last"会影响可以接受的input的形状
data_format官方文档
由上图可见默认是batch, width, height, channels