Tensorflow和Pytorch通道維度總結

Caffe 的通道順序是NCHW; Tensorflow的通道順序默認是NHWC（但可以設置成NCHW），NHWC 的訪存局部性更好（每三個輸入像素即可得到一個輸出像素），NCHW 則必須等所有通道輸入準備好才能得到最終輸出結果，需要佔用較大的臨時空間。 TensorFlow 爲什麼選擇 NHWC 格式作爲默認格式？因爲早期開發都是基於 CPU，使用 NHWC 比 NCHW 稍快一些（不難理解，N

>>阅读原文<<