jax.numpy.concat

Contents

jax.numpy.concat#

jax.numpy.concat(arrays, /, *, axis=0)[source]#
Parameters:
  • arrays (Sequence[jax.typing.ArrayLike])

  • axis (int | None)

Return type:

Array