def jacobian_h_x(self, inputs): h, act_grad = self.act_grads(inputs) jacobian = self.hidden.W * act_grad.dimshuffle(0, 'x', 1) return (h, T.reshape(jacobian, newshape=(self.nhid, self.nvis)))