http://blog.csdn.net/jningwei/article/details/78156011
区别
NHWC
[batch, in_height, in_width, in_channels]
NCHW
[batch, in_channels, in_height, in_width]
转换
NHWC –> NCHW:
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 1
- 2
NCHW –> NHWC:
- 1
- 2
- 3
- 4
- 5
- 6
- 7
[batch, in_height, in_width, in_channels]
[batch, in_channels, in_height, in_width]