jax.Array.compress#

abstractArray.compress(condition,axis=None,*,out=None,size=None,fill_value=0)[source]#

Return selected slices of this array along given axis.

Refer tojax.numpy.compress() for full documentation.

Parameters:

• self (Array)

• condition (ArrayLike)

• axis (int |None)

• out (None)

• size (int |None)

• fill_value (ArrayLike)

Return type:

Array