Skip to main content
Ctrl+K

flax.linen.get_logical_axis_rules#

flax.linen.get_logical_axis_rules()[source]#

Returns the global logical axis to mesh axis binding.

Return type:

Sequence[Tuple[str, Union[str, Tuple[str], None]]]