jax.Array.compress — JAX documentation (original) (raw)

• API Reference
• jax.Array.compress

Contents

• Array.compress()

jax.Array.compress#

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

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