jax.numpy.atleast_1d

目录

jax.numpy.atleast_1d#

jax.numpy.atleast_1d() → list[Array][源代码][源代码]#
jax.numpy.atleast_1d(x: ArrayLike, /) → Array
jax.numpy.atleast_1d(x: ArrayLike, y: ArrayLike, /, *arys: ArrayLike) → list[Array]

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

LAX-backend 对 numpy.atleast_1d() 的实现。

此函数的 JAX 版本在某些情况下可能会返回输入的副本，而不是视图。

原始文档字符串如下。

标量输入被转换为一维数组，而更高维度的输入则保持不变。

参数:

  • arys1 (array_like) – 一个或多个输入数组。

  • arys2 (array_like) – 一个或多个输入数组。

  • ... (array_like) – 一个或多个输入数组。

返回:

ret – 一个数组，或数组的元组，每个数组的 a.ndim >= 1。仅在必要时进行复制。

返回类型:

ndarray

目录