jax.numpy.hstack

jax.numpy.hstack(tup, dtype=None)

水平堆疊陣列。

JAX 實作的 numpy.hstack()。

對於一維或多維陣列，這等同於 jax.numpy.concatenate()，其中 axis=1。

參數：

• tup (np.ndarray | Array | Sequence[ArrayLike]) – 要堆疊的陣列序列；除了第二軸之外，每個陣列都必須具有相同的形狀。輸入陣列將被提升到至少秩 1。如果給定單個陣列，它將被視為等同於 tup = unstack(tup)，但實作將避免顯式解堆疊。

• dtype (DTypeLike | None | None) – 結果陣列的可選 dtype。如果未指定，則將通過類型提升語意中描述的類型提升規則來確定 dtype。

傳回：

堆疊的結果。

傳回類型：

Array

參見

• jax.numpy.stack()：沿任意軸堆疊

• jax.numpy.concatenate()：沿現有軸串聯。

• jax.numpy.vstack()：垂直堆疊，即沿軸 0。

• jax.numpy.dstack()：深度堆疊，即沿軸 2。

範例

純量值

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

1D 陣列

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

2D 陣列

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