jax.lax.concatenate
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.