Pack

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√

算子功能：实现张量的维度堆叠（Stack）操作，按照指定的维度进行张量的堆叠。要求输入的两个张量是形状是相同的，如输入selfX,selfY是shape为[2, 3,4, 5]的tensor，dim为1（可取0,1,2,3,4），则输出是shape为[2,2,3,4,5]的tensor；如果dim为3，则输出是shape为[2,3,4,2,5]的tensor。

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入张量	需要进行维度堆叠的输入张量。	见下方	ND
y	输入张量	需要进行维度堆叠的输入张量。	见下方	ND
out	输出	维度为4维，shape由dim和原selfx的shape共同决定，dtype需要与selfx一致。	同x	ND

无

调用方式	样例代码	说明
aclnn接口	test_pack_v2	通过[aclnnPackV2]接口方式调用pack_v2算子。