channel_first 和channel_last

以三通道为例

1、channel_first (channel, height,width)

设上图为（3,2,2）

解释：三个通道（3层）每个通道对应一个二维数组，即上图中的一层

黄：（（1,2），

          （3,4））

绿：（（5,6），

           （7,8））

蓝：（（9,10），

           （11,12））

 

2、channel_last (height,width,channel)

上图为（2,2,3）

解释：先构建一个2*2的数组，数组中每个元素不是一个数值，而是3个数值，即三个通道的值。

（【1,5,9】，【2,6,10】）

（【3,7,11， 【4,8,12】）

 

3.总结

channel_first:先分层，每层一个二维数组，最终的array是包含三个二维数组的3维数组。

channel_last: 不分层，把三层看作一个整体，但是这个整体的每个元素不再是一个值，而是3个值。