apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-08 03:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

for loop speedup in grdients X

Browse source
This commit is contained in:
James Hensman 2014-09-16 13:20:38 +01:00
parent ed754823be
commit ff63617286
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
GPy/kern/_src/stationary.py
View file

@ -171,7 +171,8 @@ class Stationary(Kern):
#the lower memory way with a loop
ret = np.empty(X.shape, dtype=np.float64)
[np.sum(tmp*(X[:,q][:,None]-X2[:,q][None,:]), axis=1, out=ret[:,q]) for q in xrange(self.input_dim)]
for q in xrange(self.input_dim):
np.sum(tmp*(X[:,q][:,None]-X2[:,q][None,:]), axis=1, out=ret[:,q])
ret /= self.lengthscale**2
return ret