enum class DataLayout

说明

枚举类型，数据排布格式。

定义

enum class DataLayout
{
    UNKNOWN = -1,
    NCHW = 0,
    NHWC = 1,
    ND = 2,
    NC1HWC0 = 3,
    FRACTAL_Z = 4,
    NC1HWC0_C04 = 12,
    HWCN = 16,
    NDHWC = 27,
    FRACTAL_NZ = 29,
    NCDHW = 30,
    NDC1HWC0 = 32,
    Z_3D = 33
}

成员

成员名称	描述
UNKNOWN	未知格式，默认值。
NCHW	NCHW格式。
NHWC	NHWC格式。
ND	表示支持任意格式，仅有Square、Tanh等这些单输入对自身处理的算子外，其它需要慎用。
NC1HWC0	5维数据格式。其中，C0与微架构强相关，该值等于cube单元的size，例如16；C1是将C维度按照C0切分：C1=C/C0，若结果不整除，最后一份数据需要padding到C0。
FRACTAL_Z	卷积的权重的格式。
NC1HWC0_C04	5维数据格式。其中，C0固定为4，C1是将C维度按照C0切分：C1=C/C0，若结果不整除，最后一份数据需要padding到C0。当前版本不支持。
HWCN	HWCN格式。
NDHWC	NDHWC格式，对于3维图像就需要使用带D（Depth）维度的格式。
FRACTAL_NZ	内部格式，用户目前无需使用。
NCDHW	对于3维图像就需要使用带D（Depth）维度的格式。
NDC1HWC0	6维数据格式，相比于NC1HWC0，仅多了D（Depth）维度。
Z_3D	3D卷积权重格式，例如Conv3D/MaxPool3D/AvgPool3D这些算子都需要这种格式来表达。

父主题： 数据结构参考