9587d62e956d178220d345ed16cab42263d8e971,qucumber/nn_states/complex_wavefunction.py,ComplexWaveFunction,__init__,#ComplexWaveFunction#Any#Any#Any#Any#,44

Before Change

        self.unitary_dict = unitary_dict if unitary_dict else unitaries.create_dict()
        self.unitary_dict = {
            k: v.to(device=self.device) for k, v in self.unitary_dict.items()
        }

    @property

After Change

    :type module: qucumber.rbm.BinaryRBM
    

    _rbm_am = None
    _rbm_ph = None
    _device = None

    def __init__(
        self, num_visible, num_hidden=None, unitary_dict=None, gpu=True, module=None
    ):
        if gpu and torch.cuda.is_available():
            warnings.warn(

Instances