Conact

产品支持情况

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

功能说明

  • 算子功能:实现张量的指定维度拼接(concatenate)操作,按照指定的维度拼接两个张量。如输入selfX,selfY是shape为[2, 3, 4,5]和[2,4,4,5]的tensor(要求两个张量的除了选择的dim的长度可以不同之外其他dim的长度相同),dim为1,则输出是shape为[2,7,4,5]的tensor。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入张量 需要进行拼接的输入张量。 见下方 ND
y 输入张量 需要进行拼接的输入张量。 见下方 ND
dim 指定维度 需要进行拼接的指定维度。 INT32 -
out 输出 维度为4维,shape由dims和原selfx的shape共同决定,dtype需要与selfx一致。 同x ND
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件:数据类型支持FLOAT、FLOAT16

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_concat_v2 通过[aclnnConcatV2]接口方式调用concat_v2算子。