jax.numpy.vsplit

jax.numpy.vsplit(ary, indices_or_sections)

將陣列垂直分割成子陣列。

numpy.vsplit() 的 JAX 實作。

請參閱 jax.numpy.split() 的文件以了解詳細資訊；vsplit 等同於 split 且 axis=0。

範例

一維陣列

>>> x = jnp.array([1, 2, 3, 4, 5, 6])
>>> x1, x2 = jnp.vsplit(x, 2)
>>> print(x1, x2)
[1 2 3] [4 5 6]

二維陣列

>>> x = jnp.array([[1, 2, 3, 4],
...                [5, 6, 7, 8]])
>>> x1, x2 = jnp.vsplit(x, 2)
>>> print(x1, x2)
[[1 2 3 4]] [[5 6 7 8]]

另請參閱

• jax.numpy.split()：沿著任何軸分割陣列。

• jax.numpy.hsplit()：水平分割，即沿著 axis=1

• jax.numpy.dsplit()：深度分割，即沿著 axis=2

• jax.numpy.array_split()：類似於 split，但允許 indices_or_sections 為無法均勻分割陣列大小的整數。

參數:

• ary (ArrayLike)

• indices_or_sections (int | Sequence[int] | ArrayLike)

回傳類型:

list[Array]