Reshape

产品支持情况

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

功能说明

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

参数说明

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

约束说明

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

调用说明

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