zoukankan      html  css  js  c++  java
  • 读懂NCHW和NHWC

    下图很好的描述了各种不同的格式。

    N代表数量, C代表channel,H代表高度,W代表宽度。

    NCHW其实代表的是[W H C N],第一个元素是000,第二个元素是沿着w方向的,即001,这样下去002 003,再接着呢就是沿着H方向,即004 005 006 007...这样到019后,沿C方向,轮到了020,之后021 022 ...一直到319,然后再沿N方向。

    NHWC的话以此类推,代表的是[C W H N],第一个元素是000,第二个沿C方向,即020,040, 060..一直到300,之后沿W方向,001 021 041 061...301..到了303后,沿H方向,即004 024 .。。304.。最后到了319,变成N方向,320,340....

    图中所示a就是指四个维度。

    那么在不同的硬件加速的情况下,选用的类型不同,在intel GPU加速的情况下,因为GPU对于图像的处理比较多,希望在访问同一个channel的像素是连续的,一般存储选用NCHW,这样在做CNN的时候,在访问内存的时候就是连续的了,比较方便。

    参考链接:

    [1] Intel MKL github  https://intel.github.io/mkl-dnn/understanding_memory_formats.html 

  • 相关阅读:
    Nginx-php-mysql
    Smokeping配置调整
    Smokeping外置邮箱告警
    部署Smokeping
    Centos7部署cacti
    通过yum安装最新服务
    Cacti部署
    MySQL磁盘写入策略以及数据安全性的相关参数
    linux中一些特殊的中文文件不能删除问题
    Slave_SQL_Running:No的两种解决办法
  • 原文地址:https://www.cnblogs.com/sunny-li/p/9630305.html
Copyright © 2011-2022 走看看