jax.numpy.atleast_1d#

jax.numpy.atleast_1d(*arys)[源代码]#

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

JAX 对 numpy.atleast_1d() 的实现。

参数:
  • arguments. (零个多个类数组)

  • arys (ArrayLike)

返回值:

对应于输入值的数组或数组列表。形状为 () 的数组将转换为形状 (1,),而其他形状的数组将保持不变返回。

返回类型:

Array | list[Array]

示例

标量参数将转换为 1D、长度为 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)]