jax.numpy.atleast_1d

jax.numpy.atleast_1d(*arys)

将输入转换为至少具有 1 维的数组。

JAX 实现的 numpy.atleast_1d()。

参数:

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

• arys (类数组)

返回:

一个或多个数组的列表，对应于输入值。形状为 () 的数组会被转换为形状为 (1,) 的数组，而其他形状的数组则保持不变。

返回类型:

Array | list[Array]

另请参阅

• jax.numpy.asarray()

• jax.numpy.atleast_2d()

• jax.numpy.atleast_3d()

示例

标量参数会被转换为 1 维、长度为 1 的数组

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

更高维度的输入会保持不变

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

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

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