jax.numpy.hstack#
- jax.numpy.hstack(tup, dtype=None)[source]#
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
- Returns
stacked – The array formed by stacking the given arrays.
- Return type
ndarray