pytc.kmat.calc_K3_kernel

pytc.kmat.calc_K3_kernel(xi_phi_r1, xi_phi_r2, weights_r1, weights_r2, jastrow_factor, jastrow_params, grid_r1, grid_r2, batch_size=1024)[source]

Calculate K3 kernel: K3_{kl} = sum_{g,h} w_g w_h xi_phi(k,g) |nabla u(g,h)|^2 xi_phi(l,h)

Parameters:

  • xi_phi_r1 – (N_fused, N_grid_r1) - Host resident

  • xi_phi_r2 – (N_fused, N_grid_r2) - Host resident

  • weights_r1 – (N_grid_r1,)

  • weights_r2 – (N_grid_r2,)

  • jastrow_factor – JAX Jastrow factor instance

  • jastrow_params – Parameters for the Jastrow factor

  • grid_r1 – (N_grid_r1, 3)

  • grid_r2 – (N_grid_r2, 3)

  • batch_size – Batch size for grid integration

Returns:

(N_fused, N_fused)

Return type:

K3_kernel