jax.Array.compress

abstract Array.compress(condition, axis=None, *, out=None, size=None, fill_value=0)

返回此数组沿给定轴的选定切片。

有关完整文档，请参阅jax.numpy.compress()。

参数:

• self (Array)

• condition (ArrayLike)

• axis (int | None)

• out (None)

• size (int | None)

• fill_value (类数组对象)

返回类型:

数组