DataFormatDimMap

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：根据源数据格式和目标数据格式的维度映射关系，将输入的维度索引转换为目标格式下的对应维度索引。
• 计算示例：输入维度索引[0, 1, 2, 3]，源格式NHWC，目标格式NCHW，输出[0, 2, 3, 1]。即NHWC的第1维(H)对应NCHW的第2维，第2维(W)对应第3维，第3维(C)对应第1维。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	维度索引张量，取值范围为 [-(格式长度),格式长度)。	INT32、INT64	ND
src_format	可选属性	源数据格式字符串，如"NHWC"。 默认值为"NHWC"。	STRING	-
dst_format	可选属性	目标数据格式字符串，如"NCHW"。 默认值为"NCHW"。	STRING	-
y	输出	映射后的维度索引张量，shape与输入x相同。	INT32、INT64	ND

约束说明

• src_format和dst_format必须包含相同的字符集（即相同的维度字母，顺序可不同）。
• 格式字符串长度支持1~5。
• 输入x的shape维度数不超过8。

调用说明

调用方式	调用样例	说明
图模式	test_geir_data_format_dim_map	通过算子IR构图方式调用DataFormatDimMap算子。