jax.numpy.atleast_2d

内容

jax.numpy.atleast_2d#

jax.numpy.atleast_2d() → list[Array][source]#
jax.numpy.atleast_2d(x: ArrayLike, /) → Array
jax.numpy.atleast_2d(x: ArrayLike, y: ArrayLike, /, *arys: ArrayLike) → list[Array]

将输入转换为至少具有 2 个维度的数组。

JAX 实现 numpy.atleast_2d().

参数：:

参数。 (零个或多个类数组)

返回值:

与输入值相对应的数组或数组列表。形状为 () 的数组将转换为形状 (1, 1)，形状为 (N,) 的一维数组将转换为形状 (1, N)，其他所有形状的数组将保持不变。

另请参阅

示例

标量参数将转换为二维、大小为 1 的数组

>>> x = jnp.float32(1.0)
>>> jnp.atleast_2d(x)
Array([[1.]], dtype=float32)

一维参数在形状前附加一个单位维度

>>> y = jnp.arange(4)
>>> jnp.atleast_2d(y)
Array([[0, 1, 2, 3]], dtype=int32)

更高维的输入保持不变

>>> z = jnp.ones((2, 3))
>>> jnp.atleast_2d(z)
Array([[1., 1., 1.],
       [1., 1., 1.]], dtype=float32)

可以将多个参数一次传递给函数，在这种情况下，将返回结果列表

>>> jnp.atleast_2d(x, y)
[Array([[1.]], dtype=float32), Array([[0, 1, 2, 3]], dtype=int32)]
内容