jax.numpy.trim_zeros

内容

jax.numpy.trim_zeros#

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

修剪输入数组的前导和/或尾随零。

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