文件最后提交记录最后更新时间
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
README

ConcatD

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

功能说明

  • 算子功能:将tensors中所有tensor按照维度dim进行级联,除了dim对应的维度以外的维度必须一致。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 需要级联的tensor列表。 FLOAT、FLOAT16、INT32、INT64、INT16、INT8、UINT8、BOOL、BFLOAT16、DOUBLE、COMPLEX64 ND
concat_dim 属性 需要级联的维度。 INT -
N 属性 指定要级联的tensor个数。 INT -
y 输出 输出tensor。 FLOAT、FLOAT16、INT32、INT64、INT16、INT8、UINT8、BOOL、BFLOAT16、DOUBLE、COMPLEX64 ND

约束说明

  • x列表中元素的数据类型和数据格式不在支持的范围之内。
  • x列表中无法做数据类型推导。
  • 推导出的数据类型无法转换为指定输出y的类型。
  • 非级联维度shape不一致。
  • dim超过x维度范围。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_cat 通过aclnnCat接口方式调用ConcatD算子。