Skip to contents

Bind mlx arrays along an axis

Source: R/concat.R
abind.Rd

Bind mlx arrays along an axis

Usage

abind(..., along = 1L)

Arguments

...

One or more arrays (or a single list of arrays) coercible via as_mlx().

along

Positive integer giving the existing axis (1-indexed) along which to bind the inputs.

Value

An mlx array formed by concatenating the inputs along along.

Details

This is an MLX-backed alternative to abind::abind(). All inputs must share the same shape on non-bound axes. The along axis must already exist; to create a new axis use mlx_stack().

See also

mlx.core.concatenate

Examples

x <- mlx_array(1:12, c(2, 3, 2))
y <- mlx_array(13:24, c(2, 3, 2))
z <- abind(x, y, along = 3)
dim(z)
#> [1] 2 3 4