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 (ArrayLike)

返回類型:

Array