updates.update(obj_optimizer([d_obj_padams], self.obj_params))
def __call__(self, z):
return self.op.apply(self.tf)
class KSD(Operator):
R
After Change
params = self.test_params + kwargs["more_tf_params"]
grad *= -1
grad = theano.clone(grad, {op.input_matrix: z})
grad = tt.grad(None, params, known_grads={z: grad})
return grad
class KSD(Operator):