ConcatV2

产品支持情况

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

功能说明

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

  • 输入：

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

  • 流程：

    1. 校验所有输入 Tensor 数据类型一致；
    2. 校验除 concat_dim 外所有维度完全相同；
    3. 沿 concat_dim 维度依次拼接： y = ConcatV2(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、QINT8、QUINT8、QINT32、UINT16、COMPLEX128、UINT32、UINT64、QINT16、QUINT16、BOOL、STRING、HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0	ND
concat_dim	输入	指定拼接维度，即计算流程中的 concat_dim。	INT32、INT64	ND
N	可选属性	输入 x 的数量，默认值为 1。	INT	-
y	输出	计算流程中的输出 y。	BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT32、UINT8、INT16、INT8、COMPLEX64、INT64、QINT8、QUINT8、QINT32、UINT16、COMPLEX128、UINT32、UINT64、QINT16、QUINT16、BOOL、STRING、HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, FLOAT8_E8M0	ND

• Atlas 训练、推理系列产品：不支持BFLOAT16。

约束说明

• 所有输入 Tensor 在除拼接维度外的形状必须一致。
• 输入列表 "x" 至少包含 2 个 Tensor。
• 拼接维度 concat_dim 必须在输入 Tensor 的合法维度范围内。
• x 中所有 Tensor 数据类型必须一致。
• 属性 N 指定输入数量，仅用于描述，不影响运行时动态输入列表。

调用说明

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