jax.Array.compress

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

Return selected slices of this array along given axis.

Refer to jax.numpy.compress() for full documentation.

Parameters:

• self (Array)

• condition (ArrayLike)

• axis (int | None)

• out (None)

• size (int | None)

• fill_value (ArrayLike)

Return type:

Array