l steps. This is useful to save a large amount of memory and to allow
the processing of larger images.

forward

<
source
>
(
sample: FloatTensor
sample_posterior: bool = False
return_dict: bool = True
generator: typing.Optional[torch._C.Generator] = None

)


Parameters 

sample (torch.FloatTensor) — Input sample.


sample_posterior (bool, optional, defaults to False) —
Whether to sample from the posterior.


return_dict (bool, optional, defau