Kernel Class Reference

#include <kernel.h>

Collaboration diagram for Kernel:

Public Member Functions
	Kernel ()
void	set_para (const int kernel_type_in, const double scaling_in, const double yukawa_alpha_in, const std::string &kernel_file_in)
void	fill_kernel (const int fftdim, const int ndata, const torch::Tensor &rho, const double *volume, const std::string *cell, const torch::Device device, const std::vector< torch::Tensor > &fft_gg)
double	wt_kernel (double eta, double tf_weight=1., double vw_weight=1.)
double	yukawa_kernel (double eta, double alpha)
void	read_kernel (const int fftdim, const int ndata, const torch::Tensor &rho, const double *volume, const std::string *cell, const torch::Device device, const std::vector< torch::Tensor > &fft_gg)

Public Attributes
int	kernel_type = 0
double	scaling = 0.
double	yukawa_alpha = 0.
std::string	kernel_file = "none"
std::vector< torch::Tensor >	kernel

Constructor & Destructor Documentation

Kernel()

Kernel::Kernel ( )

inline

Member Function Documentation

fill_kernel()

void Kernel::fill_kernel	(	const int	fftdim,
		const int	ndata,
		const torch::Tensor &	rho,
		const double *	volume,
		const std::string *	cell,
		const torch::Device	device,
		const std::vector< torch::Tensor > &	fft_gg
	)

Here is the call graph for this function:

Here is the caller graph for this function:

read_kernel()

void Kernel::read_kernel	(	const int	fftdim,
		const int	ndata,
		const torch::Tensor &	rho,
		const double *	volume,
		const std::string *	cell,
		const torch::Device	device,
		const std::vector< torch::Tensor > &	fft_gg
	)

Here is the caller graph for this function:

set_para()

void Kernel::set_para ( const int  kernel_type_in, const double  scaling_in, const double  yukawa_alpha_in, const std::string &  kernel_file_in  )

inline

Here is the caller graph for this function:

wt_kernel()

double Kernel::wt_kernel	(	double	eta,
		double	tf_weight = 1.,
		double	vw_weight = 1.
	)

Here is the caller graph for this function:

yukawa_kernel()

double Kernel::yukawa_kernel	(	double	eta,
		double	alpha
	)

Here is the caller graph for this function:

Member Data Documentation

kernel

std::vector<torch::Tensor> Kernel::kernel

kernel_file

std::string Kernel::kernel_file = "none"

kernel_type

int Kernel::kernel_type = 0

scaling

double Kernel::scaling = 0.

yukawa_alpha

double Kernel::yukawa_alpha = 0.

---

The documentation for this class was generated from the following files:

• /home/runner/work/abacus-develop/abacus-develop/source/source_pw/module_ofdft/ml_tools/kernel.h
• /home/runner/work/abacus-develop/abacus-develop/source/source_pw/module_ofdft/ml_tools/kernel.cpp