[slicing] fixed slicing for second order derivatives

This commit is contained in:
mzwiessele 2016-04-22 15:46:30 +01:00
parent fd5d9348d1
commit af286ba528
8 changed files with 250 additions and 82 deletions

View file

@ -378,9 +378,9 @@ class GP(Model):
dK_dXnew_full[i] = kern.gradients_X(one, Xnew, self._predictive_variable[[i]])
if full_cov:
dK2_dXdX = kern.gradients_XX(one, Xnew)
dK2_dXdX = kern.gradients_XX(one, Xnew, cov=False)
else:
dK2_dXdX = kern.gradients_XX_diag(one, Xnew)
dK2_dXdX = kern.gradients_XX_diag(one, Xnew, cov=False)
def compute_cov_inner(wi):
if full_cov: