10bdd46d76d35cfba3a484b007ea3e14af801813,qucumber/nn_states/positive_wavefunction.py,PositiveWaveFunction,psi,#PositiveWaveFunction#Any#,122

Before Change

        amplitude = self.amplitude(v)

        // complex vector; shape: (2, len(v))
        psi = torch.zeros((2,) + amplitude.shape).to(
            dtype=torch.double, device=self.device
        )
        psi[0] = amplitude
        return psi

    def gradient(self, v):
        rCompute the gradient of the effective energy for a batch of states.

After Change

        :rtype: torch.Tensor
        
        // vector/tensor of shape (2, len(v))
        return cplx.make_complex(self.amplitude(v))

    def gradient(self, v):
        rCompute the gradient of the effective energy for a batch of states.

Instances