erwise a tuple.
When returning a tuple, the first element is the sample tensor.


Predict the sample at the previous timestep via the reverse transition distribution i.e. Equation (11). See the
docstring for self.q_posterior for more in depth docs on how Equation (11) is computed.
