Fixed do_test_latents(self, Y) to use self.output_dim instead of input_dim