Caffe、TensorFlow及Pytorch通道维度顺序

发表于 2021-10-27 更新于 2022-01-25 分类于 DL Framework ， deep learning ， object detection. 阅读次数：

Caffe：NCHW
TensorFlow：默认为NHWC（但可以设置为NCHW）, NHWC的访存局部性更好（cache利用率高），NCHW需要等所有通道输入准备好才能得到最终输出结果，需要占用较大的临时空间。
Pytorch：NCHW
NCHW为NVIDIA Cudnn默认格式，使用GPU加速时用NCHW格式速度大部分情况下会更快。