flax.linen.LogicallyPartitioned
===============================

.. currentmodule:: flax.linen

.. autoclass:: LogicallyPartitioned

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~LogicallyPartitioned.__init__
      ~LogicallyPartitioned.add_axis
      ~LogicallyPartitioned.get_partition_spec
      ~LogicallyPartitioned.get_sharding
      ~LogicallyPartitioned.remove_axis
      ~LogicallyPartitioned.replace
      ~LogicallyPartitioned.replace_boxed
      ~LogicallyPartitioned.unbox
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~LogicallyPartitioned.mesh
      ~LogicallyPartitioned.rules
      ~LogicallyPartitioned.value
      ~LogicallyPartitioned.names