tiny\_diff.models.vae.model.VAE
===============================

.. currentmodule:: tiny_diff.models.vae.model

.. autoclass:: VAE
   :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::
   
      ~VAE.__init__
      ~VAE.add_module
      ~VAE.apply
      ~VAE.bfloat16
      ~VAE.buffers
      ~VAE.children
      ~VAE.compile
      ~VAE.cpu
      ~VAE.cuda
      ~VAE.double
      ~VAE.eval
      ~VAE.extra_repr
      ~VAE.float
      ~VAE.forward
      ~VAE.forward_decode
      ~VAE.forward_encode
      ~VAE.get_buffer
      ~VAE.get_extra_state
      ~VAE.get_parameter
      ~VAE.get_submodule
      ~VAE.half
      ~VAE.ipu
      ~VAE.latent_sample
      ~VAE.load_state_dict
      ~VAE.loss
      ~VAE.loss_with_fwd
      ~VAE.modules
      ~VAE.named_buffers
      ~VAE.named_children
      ~VAE.named_modules
      ~VAE.named_parameters
      ~VAE.parameters
      ~VAE.register_backward_hook
      ~VAE.register_buffer
      ~VAE.register_forward_hook
      ~VAE.register_forward_pre_hook
      ~VAE.register_full_backward_hook
      ~VAE.register_full_backward_pre_hook
      ~VAE.register_load_state_dict_post_hook
      ~VAE.register_module
      ~VAE.register_parameter
      ~VAE.register_state_dict_pre_hook
      ~VAE.reparameterize
      ~VAE.requires_grad_
      ~VAE.sample
      ~VAE.save
      ~VAE.set_extra_state
      ~VAE.share_memory
      ~VAE.state_dict
      ~VAE.to
      ~VAE.to_empty
      ~VAE.train
      ~VAE.type
      ~VAE.xpu
      ~VAE.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~VAE.T_destination
      ~VAE.call_super_init
      ~VAE.dump_patches
      ~VAE.training