在 GPU 上训练时使用 NCHW 格式,在 CPU 上做预测时使用 NHWC 格式
https://mp.weixin.qq.com/s/I4Q1Bv7yecqYXUra49o7tw 讲的简单明了,mark