jax.lax.concatenate

Warning

This page was created from a pull request (#9655).

jax.lax.concatenate¶

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

Concatenates a sequence of arrays along dimension.

Wraps XLA’s Concatenate operator.

Parameters

  • operands (Sequence[Any]) – 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

Any

Returns

An array containing the concatenation.

previous

jax.lax.complex

next

jax.lax.conj