jax.numpy.hstack

内容

jax.numpy.hstack#

jax.numpy.hstack(tup, dtype=None)[source]#

水平堆叠数组。

JAX 实现的 numpy.hstack()。

对于一个或多个维度的数组，这等价于 jax.numpy.concatenate() 且 axis=1。

参数：

  • tup (np.ndarray | Array | Sequence[ArrayLike]) – 要堆叠的数组序列；除了第二个轴之外，每个数组都必须具有相同的形状。输入数组将被提升至至少秩 1。如果给定单个数组，它将被等效地视为 tup = unstack(tup)，但实现将避免显式拆分。

  • dtype (DTypeLike | None | None) – 结果数组的可选数据类型。如果未指定，则数据类型将通过 类型提升语义 中描述的类型提升规则确定。

返回值:

堆叠的结果。

返回类型:

数组

另请参阅

示例

标量值

>>> jnp.hstack([1, 2, 3])
Array([1, 2, 3], dtype=int32, weak_type=True)

一维数组

>>> x = jnp.arange(3)
>>> y = jnp.ones(3)
>>> jnp.hstack([x, y])
Array([0., 1., 2., 1., 1., 1.], dtype=float32)

二维数组

>>> x = x.reshape(3, 1)
>>> y = y.reshape(3, 1)
>>> jnp.hstack([x, y])
Array([[0., 1.],
       [1., 1.],
       [2., 1.]], dtype=float32)
内容