flax.linen.logical_to_mesh# flax.linen.logical_to_mesh(tree, rules=None)[source]# Applies logical_to_mesh_axes to pytrees of logical PartitionSpecs. Parameters: tree (Any) – rules (Optional[Sequence[Tuple[str, Union[str, Tuple[str], None]]]]) – Return type: Any