def gen_samples(self, z0): samples = self.model.gen_samples(z0=z0) if self.nc == 1: samples = np.tile(samples, [1,1,1,3]) return samples