jax.numpy.trim_zeros

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

修剪输入数组的开头和/或结尾的零。

JAX 实现的 numpy.trim_zeros()。

参数:

• filt (类似数组) – 输入数组。必须满足 filt.ndim == 1。

• trim (字符串) –

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

  • f - 仅修剪开头的零。

  • b - 仅修剪结尾的零。

  • fb - 修剪开头和结尾的零。

返回:

包含修剪后的输入的数组，其数据类型与 filt 相同。

返回类型:

数组

示例

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