jax.numpy.trim_zeros

jax.numpy.trim_zeros(filt, trim='fb')

修剪輸入陣列的開頭和/或結尾的零。

JAX 版本的 numpy.trim_zeros()。

參數:

• filt (ArrayLike) – 輸入陣列。必須有 filt.ndim == 1。

• trim (str) –

  字串，選用，預設 = fb。指定從哪個末端修剪輸入。

  • f - 僅修剪開頭的零。

  • b - 僅修剪結尾的零。

  • fb - 修剪開頭和結尾的零。

返回:

一個包含修剪後輸入的陣列，其 dtype 與 filt 相同。

返回型別:

Array

範例

>>> x = jnp.array([0, 0, 2, 0, 1, 4, 3, 0, 0, 0])
>>> jnp.trim_zeros(x)
Array([2, 0, 1, 4, 3], dtype=int32)