tiny\_diff.models.vae.components.ConvVAEComponent
=================================================

.. currentmodule:: tiny_diff.models.vae.components

.. autoclass:: ConvVAEComponent
   :members:                                    <-- add at least this line
   :show-inheritance:                           <-- plus I want to show inheritance...
   :no-inherited-members:                          <-- ...and inherited members too

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~ConvVAEComponent.__init__
      ~ConvVAEComponent.add_module
      ~ConvVAEComponent.apply
      ~ConvVAEComponent.bfloat16
      ~ConvVAEComponent.buffers
      ~ConvVAEComponent.children
      ~ConvVAEComponent.compile
      ~ConvVAEComponent.conv_forward
      ~ConvVAEComponent.conv_kwargs
      ~ConvVAEComponent.cpu
      ~ConvVAEComponent.cuda
      ~ConvVAEComponent.double
      ~ConvVAEComponent.eval
      ~ConvVAEComponent.extra_repr
      ~ConvVAEComponent.float
      ~ConvVAEComponent.forward
      ~ConvVAEComponent.get_buffer
      ~ConvVAEComponent.get_extra_state
      ~ConvVAEComponent.get_parameter
      ~ConvVAEComponent.get_submodule
      ~ConvVAEComponent.half
      ~ConvVAEComponent.ipu
      ~ConvVAEComponent.load_state_dict
      ~ConvVAEComponent.modules
      ~ConvVAEComponent.named_buffers
      ~ConvVAEComponent.named_children
      ~ConvVAEComponent.named_modules
      ~ConvVAEComponent.named_parameters
      ~ConvVAEComponent.parameters
      ~ConvVAEComponent.register_backward_hook
      ~ConvVAEComponent.register_buffer
      ~ConvVAEComponent.register_forward_hook
      ~ConvVAEComponent.register_forward_pre_hook
      ~ConvVAEComponent.register_full_backward_hook
      ~ConvVAEComponent.register_full_backward_pre_hook
      ~ConvVAEComponent.register_load_state_dict_post_hook
      ~ConvVAEComponent.register_module
      ~ConvVAEComponent.register_parameter
      ~ConvVAEComponent.register_state_dict_pre_hook
      ~ConvVAEComponent.requires_grad_
      ~ConvVAEComponent.res_kwargs
      ~ConvVAEComponent.save
      ~ConvVAEComponent.set_extra_state
      ~ConvVAEComponent.share_memory
      ~ConvVAEComponent.state_dict
      ~ConvVAEComponent.to
      ~ConvVAEComponent.to_empty
      ~ConvVAEComponent.train
      ~ConvVAEComponent.type
      ~ConvVAEComponent.xpu
      ~ConvVAEComponent.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~ConvVAEComponent.T_destination
      ~ConvVAEComponent.align_in_ch
      ~ConvVAEComponent.align_out_ch
      ~ConvVAEComponent.call_super_init
      ~ConvVAEComponent.channels
      ~ConvVAEComponent.conv_channels
      ~ConvVAEComponent.dump_patches
      ~ConvVAEComponent.embedding_size
      ~ConvVAEComponent.n_conv_layers
      ~ConvVAEComponent.output_channels
      ~ConvVAEComponent.residual_cls
      ~ConvVAEComponent.scale_factor
      ~ConvVAEComponent.stride
      ~ConvVAEComponent.training