364948bf0bb46f9bc3809bbe0a63d09348994444,qucumber/complex_wavefunction.py,ComplexWavefunction,__init__,#ComplexWavefunction#Any#Any#Any#Any#,30

Before Change

        self.networks = ["rbm_am","rbm_ph"]
        self.device = self.rbm_am.device
        self.unitary_dict = unitary_dict 
        self.visible_state = torch.zeros(1,self.num_visible,
                                         device=self.rbm_am.device,
                                         dtype=torch.double)
        self.hidden_state = torch.zeros(1,self.num_hidden,
                                         device=self.rbm_am.device,
                                         dtype=torch.double)

After Change

        self.networks = ["rbm_am", "rbm_ph"]
        self.device = self.rbm_am.device
        self.unitary_dict = {k: v.to(device=self.device)
                             for k, v in unitary_dict.items()}

    def initialize_parameters(self):
        rRandomize the parameters
        :math:`\bm{\lambda}=\{\bm{W}^{\bm{\lambda}},

Instances