jax.numpy.atleast_3d

jax.numpy.atleast_3d(*arys)

View inputs as arrays with at least three dimensions.

LAX-backend implementation of numpy.atleast_3d().

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 three 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 three or more dimensions are preserved.

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

• arys (ArrayLike) –

Returns:

res1, res2, … – An array, or list of arrays, each with a.ndim >= 3. Copies are avoided where possible, and views with three or more dimensions are returned. For example, a 1-D array of shape (N,) becomes a view of shape (1, N, 1), and a 2-D array of shape (M, N) becomes a view of shape (M, N, 1).

Return type:

ndarray