ascendie.DataLayout
说明
Tensor张量的数据排布。
成员
成员名称 |
描述 |
同义词 |
|---|---|---|
ascendie.DataLayout.unknown |
未知格式，默认值。 |
ascendie.unknown |
ascendie.DataLayout.nchw |
NCHW格式。 |
ascendie.nchw |
ascendie.DataLayout.nhwc |
NHWC格式。 |
ascendie.nhwc |
ascendie.DataLayout.nd |
表示支持任意格式，仅有Square、Tanh等这些单输入对自身处理的算子外，其它需要慎用。 |
ascendie.nd |
ascendie.DataLayout.nc1hwc0 |
5维数据格式。其中，C0与微架构强相关，该值等于cube单元的size，例如16；C1是将C维度按照C0切分：C1=C/C0， 若结果不整除，最后一份数据需要padding到C0。 |
ascendie.nc1hwc0 |
ascendie.DataLayout.fractal_z |
卷积的权重的格式。 |
ascendie.fractal_z |
ascendie.DataLayout.nc1hwc0_c04 |
5维数据格式。其中，C0固定为4，C1是将C维度按照C0切分：C1=C/C0， 若结果不整除，最后一份数据需要padding到C0。当前版本不支持。 |
ascendie.nc1hwc0_c04 |
ascendie.DataLayout.hwcn |
HWCN格式。 |
ascendie.hwcn |
ascendie.DataLayout.ndhwc |
NDHWC格式，对于3维图像就需要使用带D（Depth）维度的格式。 |
ascendie.ndhwc |
ascendie.DataLayout.fractal_nz |
内部格式，用户目前无需使用。 |
ascendie.fractal_nz |
ascendie.DataLayout.ncdhw |
对于3维图像就需要使用带D（Depth）维度的格式。 |
ascendie.ncdhw |
ascendie.DataLayout.ndc1hwc0 |
6维数据格式，相比于NC1HWC0，仅多了D（Depth）维度。 |
ascendie.ndc1hwc0 |
ascendie.DataLayout.z_3d |
3D卷积权重格式，例如Conv3D/MaxPool3D/AvgPool3D这些算子都需要这种格式来表达。 |
ascendie.z_3d |