qmctorch.wavefunction.jastrows.elec_elec_nuclei.kernels.fully_connected_jastrow_kernel module¶

class qmctorch.wavefunction.jastrows.elec_elec_nuclei.kernels.fully_connected_jastrow_kernel.FullyConnectedJastrowKernel(nup, ndown, atomic_pos, cuda)[source]¶

Bases: qmctorch.wavefunction.jastrows.elec_elec_nuclei.kernels.jastrow_kernel_electron_electron_nuclei_base.JastrowKernelElectronElectronNucleiBase

Defines a fully connected jastrow factors.

forward(x)[source]¶

Compute the values of the individual f_ij=f(r_ij)

Parameters:	x (torch.tensor) – e-e distance Nbatch, Nele_pairs
Returns:	values of the f_ij
Return type:	torch.tensor