spatialvi.module.JVAE
=====================

.. currentmodule:: spatialvi.module

.. autoclass:: JVAE

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~JVAE.__init__
      ~JVAE.add_module
      ~JVAE.apply
      ~JVAE.bfloat16
      ~JVAE.buffers
      ~JVAE.children
      ~JVAE.compile
      ~JVAE.cpu
      ~JVAE.cuda
      ~JVAE.double
      ~JVAE.eval
      ~JVAE.extra_repr
      ~JVAE.float
      ~JVAE.forward
      ~JVAE.generative
      ~JVAE.get_buffer
      ~JVAE.get_extra_state
      ~JVAE.get_parameter
      ~JVAE.get_sample_rate
      ~JVAE.get_submodule
      ~JVAE.half
      ~JVAE.inference
      ~JVAE.ipu
      ~JVAE.load_state_dict
      ~JVAE.loss
      ~JVAE.modules
      ~JVAE.mtia
      ~JVAE.named_buffers
      ~JVAE.named_children
      ~JVAE.named_modules
      ~JVAE.named_parameters
      ~JVAE.on_load
      ~JVAE.parameters
      ~JVAE.reconstruction_loss
      ~JVAE.register_backward_hook
      ~JVAE.register_buffer
      ~JVAE.register_forward_hook
      ~JVAE.register_forward_pre_hook
      ~JVAE.register_full_backward_hook
      ~JVAE.register_full_backward_pre_hook
      ~JVAE.register_load_state_dict_post_hook
      ~JVAE.register_load_state_dict_pre_hook
      ~JVAE.register_module
      ~JVAE.register_parameter
      ~JVAE.register_state_dict_post_hook
      ~JVAE.register_state_dict_pre_hook
      ~JVAE.requires_grad_
      ~JVAE.sample
      ~JVAE.sample_from_posterior_l
      ~JVAE.sample_from_posterior_z
      ~JVAE.sample_rate
      ~JVAE.sample_scale
      ~JVAE.set_extra_state
      ~JVAE.set_submodule
      ~JVAE.share_memory
      ~JVAE.state_dict
      ~JVAE.to
      ~JVAE.to_empty
      ~JVAE.train
      ~JVAE.type
      ~JVAE.xpu
      ~JVAE.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~JVAE.T_destination
      ~JVAE.call_super_init
      ~JVAE.device
      ~JVAE.dump_patches
      ~JVAE.training