Shape

产品支持情况

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

功能说明

• 算子功能：返回输入tensor的shape，即由输入各维度值组成的一维tensor。

• 计算逻辑：

  • 输入tensor的rank为r
  • 输出为一维tensor，长度为r，内容为输入tensor的各维度值

• 示例：

  输入: x = float32[2, 3, 4]
  输出: y = int32[3] = [2, 3, 4]
  

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待获取shape信息的输入tensor。	任意数据类型	ND
y	输出	输入tensor的各维度值组成的一维tensor，长度等于输入的rank。	INT32、INT64	ND
dtype	属性	输出tensor的数据类型，缺省时默认为INT32。	INT32、INT64	-

约束说明

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

调用说明

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