shape changes for gradeitns of likelihood parameters in variational_expectations

This commit is contained in:
James Hensman 2015-03-27 11:14:41 +00:00
parent e74bfd81c6
commit 55dba3d2d9
3 changed files with 6 additions and 6 deletions

View file

@ -48,7 +48,7 @@ class SVGP(LatentFunctionInference):
#rescale the F term if working on a batch
F, dF_dmu, dF_dv = F*batch_scale, dF_dmu*batch_scale, dF_dv*batch_scale
if dF_dthetaL is not None:
dF_dthetaL = dF_dthetaL.sum(1)*batch_scale
dF_dthetaL = dF_dthetaL.sum(1).sum(1)*batch_scale
#derivatives of expected likelihood
Adv = A.T[:,:,None]*dF_dv[None,:,:] # As if dF_Dv is diagonal