Concat#

Concat operation concatenates \(N\) tensors over axis (here designated \(C\)) and is defined as (the variable names follow the standard Conventions):

\[\dst(\overline{ou}, c, \overline{in}) = \src\_i(\overline{ou}, c’, \overline{in}),\]

where \(c = C_1 + .. + C_{i-1} {}_{} + c'\).

Operation Attributes#

Attribute

Name

Description

Value Type
Supported

Values
Required or

Optional

axis

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 the below index order when constructing an operation.

Inputs#

Index

Argument Name

Required or Optional

0

src_i

Required

@note At least one input tensor is required. Data types and ranks of all input tensors should match. The dimensions of all input tensors should be the same except for the dimension specified by axis attribute.

Outputs#

Index

Argument Name

Required or Optional

0

dst

Required

Supported Data Types#

Concat operation supports the following data type combinations.

Src_i

Dst

f32

f32

bf16

bf16

f16

f16