Merge pull request #613 from dtorrejo/Multi_sample_bug

Fixes the dimensions of the samples output
This commit is contained in:
Max Zwiessele 2018-03-23 16:30:57 +01:00 committed by GitHub
commit d85c9d5379
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -592,7 +592,7 @@ class GP(Model):
if self.output_dim == 1:
return sim_one_dim(m, v)
else:
fsim = np.empty((self.output_dim, self.num_data, size))
fsim = np.empty((self.output_dim, X.shape[0], size))
for d in range(self.output_dim):
if full_cov and v.ndim == 3:
fsim[d] = sim_one_dim(m[:, d], v[:, :, d])