FilesLast commitLast update
21 days ago
21 days ago
21 days ago
21 days ago
21 days ago
21 days ago
README

PhonyConcat

产品支持情况

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

功能说明

  • 算子功能:将一组张量沿 concat_dim 指定的维度拼接。PhonyConcat 的所有输入分配在同一块连续内存上,由GE计算每个输入的内存偏移,供前置自定义算子按偏移写入内存。该算子仅用于标识GE分配连续内存,不执行任何计算。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入(动态) 输入张量列表,所有输入的shape和数据类型须一致。 ALL ND
concat_dim 必选属性 拼接维度列表,取值范围为 [-len(x.shape), len(x.shape)-1],与 N 一一对应。 LIST_INT -
N 必选属性 每个拼接维度上的拼接数量列表,各元素乘积须等于输入张量个数且各元素大于等于1。 LIST_INT -
keep_input_offset 可选属性 是否计算输入张量的内存偏移,默认为 true。 BOOL -
y 输出 拼接后的输出张量,shape为输入shape在各 concat_dim 维度上乘以对应的 N 值,数据类型与输入一致。 与 x 相同 ND

约束说明

  • concat_dim 与 N 的元素个数须一致,N 各元素乘积须等于输入张量个数。
  • 所有输入张量的shape与数据类型须一致。
  • 输入张量的内存大小须32字节对齐(主机侧Host CPU执行引擎约束)。
  • 该算子不支持动态shape(unknown shape)图的编译执行。
  • 支持的数据类型以 phony_concat_proto.h 中的IR定义为准。

调用说明

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