Skip to contents

Primitive Concatenate

Source: R/primitives.R
prim_concatenate.Rd

Concatenates arrays along an axis.

Usage

prim_concatenate(..., axis)

Arguments

...

(arrayish)
Arrays to concatenate. Must all have the same data type, naxes, and shape except along axis.

axis

(integer(1))
Axis along which to concatenate. Negative values count from the end, i.e. -1 refers to the last axis.

Value

arrayish
Has the same data type as the inputs. The output shape matches the inputs in all axes except axis, which is the sum of the input sizes along that axis. It is ambiguous if all inputs are ambiguous.

Implemented Rules

  • stablehlo

  • quickr

  • reverse

StableHLO

Lowers to hlo_concatenate().

See also

nv_concatenate()

Examples

x <- nv_array(c(1, 2, 3))
y <- nv_array(c(4, 5, 6))
prim_concatenate(x, y, axis = 1L)
#> AnvlArray
#>  1
#>  2
#>  3
#>  4
#>  5
#>  6
#> [ CPUf32{6} ]