Maintains consistency with numpy arrays

This commit is contained in:
Diego Torrejon 2018-03-22 16:32:05 -04:00
parent 90c2912ace
commit 69eb888ad1

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, len(X), 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])