Pack

产品支持情况

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

功能说明

• 算子功能：通过沿轴维度打包，将值中的张量列表打包成一个比值中的每个张量高一维度的张量。给定形状为（A，B，C）的张量的长度为N的列表；如果轴==0，则输出张量将具有形状（N，A，B，C）。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	需要级联的tensor列表。	COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND
axis	可选属性	指定沿其打包的维度。	INT	-
N	可选属性	指定要打包的tensor个数。	INT	-
y	输出	输出结果。	COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND

• Atlas训练系列产品、Atlas推理系列产品、Atlas 200I/500 A2推理产品：不支持BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN。
• Atlas A2系列产品、Atlas A3系列产品：不支持HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN。

约束说明

无。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_stack	通过aclnnStack接口方式调用Pack算子。