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.