jax.numpy.hstack

jax.numpy.hstack(tup, dtype=None)

Stack arrays in sequence horizontally (column wise).

LAX-backend implementation of numpy.hstack().

Original docstring below.

This is equivalent to concatenation along the second axis, except for 1-D arrays where it concatenates along the first axis. Rebuilds arrays divided by hsplit.

This function makes most sense for arrays with up to 3 dimensions. For instance, for pixel-data with a height (first axis), width (second axis), and r/g/b channels (third axis). The functions concatenate, stack and block provide more general stacking and concatenation operations.

Parameters

• tup (sequence of ndarrays) – The arrays must have the same shape along all but the second axis, except 1-D arrays which can be any length.

• dtype (Union[Any, str, dtype, SupportsDType, None]) –

Returns

stacked – The array formed by stacking the given arrays.

Return type

ndarray