jax.numpy.atleast_2d#
- jax.numpy.atleast_2d(*arys)[源代码]#
将输入转换为至少具有 2 个维度的数组。
numpy.atleast_2d()
的 JAX 实现。- 参数:
arguments. (零个或多个类似数组)
arys (ArrayLike)
- 返回:
对应于输入值的数组或数组列表。形状为
()
的数组被转换为形状(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)]