ShapeN

产品支持情况

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

功能说明

• 算子功能：返回多个输入tensor的shape，每个输入对应一个输出，输出为由对应输入各维度值组成的一维tensor。

• 计算逻辑：

  • 输入N个tensor，第i个tensor的rank为r_i
  • 输出为N个一维tensor，第i个输出长度为r_i，内容为第i个输入的各维度值

• 示例：

  输入: x0 = float32[2, 3], x1 = float32[4, 5, 6]
  输出: y0 = int32[2] = [2, 3], y1 = int32[3] = [4, 5, 6]
  

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	动态输入	待获取shape信息的输入tensor列表。	DOUBLE、FLOAT、FLOAT16、INT8、INT16、UINT8、UINT16、INT32、UINT32、INT64、UINT64、BOOL、STRING	ND
y	动态输出	与输入一一对应，每个输出为对应输入各维度值组成的一维tensor，长度等于对应输入的rank。	INT32、INT64	ND
dtype	属性	输出tensor的数据类型，缺省时默认为INT32。	INT32、INT64	-

约束说明

• dtype属性仅支持INT32、INT64，取其他值时算子推理失败。
• 任一输入为未知秩（-2）时，对应输出为一维未知形状（-1）。
• 输入仅支持ND格式。

调用说明

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