rbf kernel gpu implementation in progress

This commit is contained in:
Zhenwen Dai 2014-06-20 11:56:21 +01:00
parent 0b75aa8b0f
commit db9b9bc857
4 changed files with 467 additions and 8 deletions

View file

@ -73,7 +73,7 @@ class TruncLinear(Kern):
return XX
def Kdiag(self, X):
return self.variances*(np.square(X-self.delta)).sum(axis=-1)
return (self.variances*np.square(X-self.delta)).sum(axis=-1)
def update_gradients_full(self, dL_dK, X, X2=None):
dK_dvar = self._product(X, X2)