flax.linen.get_logical_axis_rules

flax.linen.get_logical_axis_rules#

flax.linen.get_logical_axis_rules()[source]#

Returns the global logical axis to mesh axis binding.