flax.linen.get_logical_axis_rules# flax.linen.get_logical_axis_rules()[source]# Returns the global logical axis to mesh axis binding.
