ConcatDV2

产品支持情况

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

功能说明

• 算子功能：用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor。
• 计算流程：

  • 输入：

    • Tensor 列表 x[0], x[1], …, x[N-1]
    • 拼接维度 concat_dim

  • 流程：

    1. 校验所有输入 Tensor 数据类型一致；
    2. 校验除 concat_dim 外所有维度完全相同；
    3. 沿 concat_dim 维度依次拼接： y = ConcatDV2(x[0], x[1], ..., x[N-1], axis = concat_dim)

  • 输出：拼接后的 Tensor y

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	动态输入列表，流程图中的输入 x[i]。	BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT32、UINT8、INT16、INT8、COMPLEX64、INT64、BOOL	ND
concat_dim	输入	指定拼接维度，即计算流程中的 concat_dim。	INT32、INT64	ND
y	输出	计算流程中的输出 y。	BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT32、UINT8、INT16、INT8、COMPLEX64、INT64、BOOL	ND

约束说明

• 所有输入 Tensor 在除拼接维度外的形状必须一致。
• 拼接维度 concat_dim 当前仅支持为0的情况。
• x 中所有 Tensor 数据类型必须一致。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_cat	通过aclnnCat接口方式调用ConcatDV2算子。