Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
Concat
General
Concat operation concatenates
tensors over axis (here designated
) and is defined as (the variable names follow the standard Naming Conventions):
where
.
Operation attributes
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
|---|---|---|---|---|
Specifies dimension along which concatenation happens. |
s64 |
A s64 value in the range of [-r, r-1] where r = rank(src) |
Required |
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
|---|---|---|
0 |
src_i |
Required |
Outputs
Index |
Argument Name |
Required or Optional |
|---|---|---|
0 |
dst |
Required |
Supported data types
Concat operation supports the following data type combinations.
Src |
Dst |
|---|---|
f32 |
f32 |
bf16 |
bf16 |
f16 |
f16 |