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)