docs: def clone(self, name=None):
      """Returns a cloned `Linear` module.
  
      Args:
        name: Optional string assigning name of cloned module. The default name
            is constructed by appending "_clone" to `self.module_name`.
  
      Returns:
        Cloned `Linear` module.
      """
      if name is None:
        name = self.module_name + "_clone"
      return Linear(output_size=self.output_size,
                    use_bias=self._use_bias,
                    initializers=self._initializers,
                    partitioners=self._partitioners,
                    regularizers=self._regularizers,
                    name=name)