jax.numpy.trim_zeros#
- jax.numpy.trim_zeros(filt, trim='fb')[source]#
修剪输入数组的前导和/或尾随零。
JAX 实现的
numpy.trim_zeros()
。- 参数:
filt (ArrayLike) – 输入数组。必须有
filt.ndim == 1
。trim (str) –
字符串,可选,默认值 =
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)