功能
Tensor拼接功能。
图1 ConcatOperation
约束
输入x和y的维度数要求一致。
输入x或y的维度大小，除了concatDim维外，其他维度要求相同。
定义
struct ConcatParam {
int concatDim = 0;
};
成员
成员名称 |
描述 |
|---|---|
concatDim |
指定拼接的维度，输入须符合x或y的维度范围。 当concatDim为负数时，其含义是从最高维度开始访问，如concatDim = -1，x维度数为dimNum，则拼接维度为dimNum - 1。 |
输入
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
y |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
输出
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
output |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |