flax.linen.Partitioned
======================

.. currentmodule:: flax.linen

.. autoclass:: Partitioned

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Partitioned.mesh
      ~Partitioned.value
      ~Partitioned.names