文件最后提交记录最后更新时间
5 个月前
3 个月前
1 个月前
28 天前
1 个月前
2 个月前
1 个月前
README

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算子。