jax.numpy.atleast_3d

目录

jax.numpy.atleast_3d#

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

将输入视为至少具有三个维度的数组。

LAX 后端实现 numpy.atleast_3d()。

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

原始文档字符串如下。

参数:

  • arys1 (array_like) – 一个或多个类似数组的序列。非数组输入将被转换为数组。已经具有三个或更多维度的数组将被保留。

  • arys2 (array_like) – 一个或多个类似数组的序列。非数组输入将被转换为数组。已经具有三个或更多维度的数组将被保留。

  • ... (array_like) – 一个或多个类似数组的序列。非数组输入将被转换为数组。已经具有三个或更多维度的数组将被保留。

返回:

res1, res2, … – 一个数组，或数组的元组，每个数组的 a.ndim >= 3。尽可能避免复制，并返回具有三个或更多维度的视图。例如，形状为 (N,) 的 1-D 数组变为形状为 (1, N, 1) 的视图，形状为 (M, N) 的 2-D 数组变为形状为 (M, N, 1) 的视图。

返回类型:

ndarray

目录