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_dimTensor 列表 x[0], x[1], …, x[N-1]流程校验所有输入 Tensor 数据类型一致校验除 concat_dim 外所有维度完全相同沿 concat_dim 维度依次拼接 y Concat(x[0], x[1], ..., x[N-1], axis concat_dim)输出拼接后的 Tensor y参数说明参数名输入/输出/属性描述数据类型数据格式concat_dim输入指定拼接维度即计算流程中的 concat_dim。INT32、INT64NDx输入动态输入列表流程图中的输入 x[i]。BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT32、UINT8、INT16、INT8、COMPLEX64、INT64、QINT8、QUINT8、QINT32、UINT16、COMPLEX128、UINT32、UINT64、QINT16、QUINT16、BOOL、STRINGNDN可选属性输入 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、STRINGNDAtlas 训练、推理系列产品不支持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创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考