76b2432b1c1ee03c80895a5c7d49cde485323102,qucumber/nn_states/density_matrix.py,DensityMatrix,rotated_gradient,#DensityMatrix#,334

Before Change

                  of the amplitude and phase RBMS
        :rtype: list[torch.Tensor, torch.Tensor]
        
        rotated_grad = [
            torch.zeros(
                getattr(self, net).num_pars, dtype=torch.double, device=self.device
            )
            for net in self.networks
        ]

        UrhoU, UrhoU_v, v = unitaries.rotate_rho_prob(
            self, basis, sample, include_extras=True
        )

After Change

        raw_grads = [self.am_grads(v, v, expand=True), self.ph_grads(v, v, expand=True)]

        rotated_grad = [
            -cplx.einsum("ij,ij...->...", UrhoU_v, g, imag_part=False)
            for g in raw_grads
        ]

        return [g / UrhoU for g in rotated_grad]

    def am_grads(self, v, vp, expand=False):

Instances