(self, z, is_init=True)
| 981 | return posterior.sample() |
| 982 | |
| 983 | def decode_naive(self, z, is_init=True): |
| 984 | z = z.to(next(self.decoder.parameters()).dtype) |
| 985 | dec = self.decoder(z, is_init) |
| 986 | return dec |
| 987 | |
| 988 | @torch.inference_mode() |
| 989 | def decode_original(self, z): |
no test coverage detected