flax.linen.with\_logical\_constraint
====================================

.. currentmodule:: flax.linen

.. autofunction:: with_logical_constraint