CANN/ops-math Concat算子

Concat

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

产品	是否支持
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

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

约束说明

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

调用说明

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

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math