jax.lax.concatenate#

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

Concatenates a sequence of arrays along dimension.

Wraps XLA’s Concatenate operator.

Parameters
Return type

Array

Returns

An array containing the concatenation.