jax.numpy.atleast_2d#

jax.numpy.atleast_2d(*arys)[source]#

View inputs as arrays with at least two dimensions.

LAX-backend implementation of numpy.atleast_2d().

The JAX version of this function may in some cases return a copy rather than a view of the input.

Original docstring below.

Parameters:

  • arys1 (array_like) – One or more array-like sequences. Non-array inputs are converted to arrays. Arrays that already have two or more dimensions are preserved.

  • arys2 (array_like) – One or more array-like sequences. Non-array inputs are converted to arrays. Arrays that already have two or more dimensions are preserved.

  • ... (array_like) – One or more array-like sequences. Non-array inputs are converted to arrays. Arrays that already have two or more dimensions are preserved.

  • arys (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

Returns:

res, res2, … – An array, or list of arrays, each with a.ndim >= 2. Copies are avoided where possible, and views with two or more dimensions are returned.

Return type:

ndarray