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 算子。 |