jax.numpy.trim_zeros#

jax.numpy.trim_zeros(filt, trim='fb')[source]#

删除输入数组开头和/或结尾的零。

numpy.trim_zeros() 的 JAX 实现。

参数:

  • filt (ArrayLike) – 输入数组。必须具有 filt.ndim == 1。

  • trim (str) –

    字符串，可选，默认 = fb。指定从哪个端修剪输入。

    • f - 仅修剪前导零。

    • b - 仅修剪尾随零。

    • fb - 修剪前导和尾随零。

返回:

一个包含修剪后的输入且具有与 filt 相同 dtype 的数组。

返回类型:

数组

示例

>>> 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)