jax.numpy.concatenate#

jax.numpy.concatenate(arrays, axis=0, dtype=None)[source]#

Join a sequence of arrays along an existing axis.

LAX-backend implementation of numpy.concatenate().

Original docstring below.

Parameters
Returns

res – The concatenated array.

Return type

ndarray