Size

产品支持情况

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

功能说明

• 算子功能：返回输入张量的尺寸，即张量元素个数的整数标量。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量。	所有类型	ND
dtype	属性	可选整数，输出数据类型，取值为DT_INT32或DT_INT64，默认为DT_INT32。	Int	-
y	输出	输入张量的元素个数，标量。	INT32, INT64	ND

约束说明

• dtype属性取值仅支持DT_INT32、DT_INT64。

调用说明

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