Reshape

产品支持情况

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

功能说明

• 算子功能：在不改变底层数据布局的前提下，将输入张量重解释为目标形状。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待重塑的输入张量。	BOOL, FLOAT, FLOAT16, INT8, INT16, UINT16, UINT8, INT32, INT64, UINT32, UINT64, DOUBLE, COMPLEX64, COMPLEX128, BF16	ND
shape	输入	目标形状张量，支持一个 -1；默认模式下 0 表示复制对应输入维度。	INT32, INT64	1D
axis	属性	起始替换维度，默认值为 0。	INT	-
num_axes	属性	被替换的连续维度数，默认值为 -1。	INT	-
y	输出	重塑后的输出张量，数据类型与输入 x 相同。	与 x 相同	ND

约束说明

• 输出元素总数必须与输入元素总数一致。
• shape 中最多只能出现一个 -1。
• 当 allowzero 未设置或为 0 时，shape 中的 0 会复制对应输入维度；当 allowzero 为 1 时，0 按字面值参与计算。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_reshape	通过 算子 IR 构图方式调用 reshape 算子。