Ctrl+K

jax.lax.concatenate

Contents

jax.lax.concatenate#

jax.lax.concatenate(operands, dimension)[source]#

Concatenates a sequence of arrays along dimension.

Wraps XLA’s Concatenate operator.

Parameters:

  • operands (Array | Sequence[ArrayLike]) – a sequence of arrays to concatenate. The arrays must have equal shapes, except in the dimension axis.

  • dimension (int) – the dimension along which to concatenate the arrays.

Return type:

Array

Returns:

An array containing the concatenation.

Contents