jax.lax.concatenate#

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

Concatenates a sequence of arrays along dimension.

Wraps XLA’s Concatenate operator.

Parameters

operands (Union[Array, Sequence[Union[Array, ndarray, bool_, number, bool, int, float, complex]]]) – 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.