Concat

产品支持情况

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

功能说明

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

  • 输入：

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

  • 流程：

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

  • 输出：拼接后的Tensor y

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
concat_dim	输入	指定拼接维度，即计算流程中的concat_dim。	INT32、INT64	ND
x	输入	动态输入列表，流程图中的输入x[i]。	BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT32、UINT8、INT16、INT8、COMPLEX64、INT64、QINT8、QUINT8、QINT32、UINT16、COMPLEX128、UINT32、UINT64、QINT16、QUINT16、BOOL、STRING	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	ND

约束说明

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

调用说明

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