mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-09 20:12:38 +02:00
more bugfix
This commit is contained in:
parent
4139397dd3
commit
3fc603f778
1 changed files with 3 additions and 10 deletions
|
|
@ -169,17 +169,10 @@ class Stationary(Kern):
|
||||||
#the lower memory way with a loop
|
#the lower memory way with a loop
|
||||||
tmp = invdist*dL_dr
|
tmp = invdist*dL_dr
|
||||||
ret = np.empty(X.shape, dtype=np.float64)
|
ret = np.empty(X.shape, dtype=np.float64)
|
||||||
<<<<<<< HEAD
|
|
||||||
[np.sum(tmp*(X[:,q:q+1]-X2[:,q:q+1]), axis=1, out=ret[:,q]) for q in xrange(self.input_dim)]
|
|
||||||
=======
|
|
||||||
if X2 is None:
|
if X2 is None:
|
||||||
[np.einsum('ij,ij->i', tmp, X[:,q][:,None]-X[:,q][None,:], out=ret[:,q]) for q in xrange(self.input_dim)]
|
tmp = tmp + tmp.T
|
||||||
ret2 = np.empty(X.shape, dtype=np.float64)
|
X2 = X
|
||||||
[np.einsum('ij,ji->j', tmp, X[:,q][:,None]-X[:,q][None,:], out=ret2[:,q]) for q in xrange(self.input_dim)]
|
[np.einsum('ij,ij->i', tmp, X[:,q][:,None]-X2[:,q][None,:], out=ret[:,q]) for q in xrange(self.input_dim)]
|
||||||
ret += ret2
|
|
||||||
else:
|
|
||||||
[np.einsum('ij,ij->i', tmp, X[:,q][:,None]-X2[:,q][None,:], out=ret[:,q]) for q in xrange(self.input_dim)]
|
|
||||||
>>>>>>> 1061bf52482aa3bf6769db810c955d5fbf51ceae
|
|
||||||
ret /= self.lengthscale**2
|
ret /= self.lengthscale**2
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue