Template Function _DP_DeepTensorCompute

Function Documentation

template<typename FPTYPE>
inline void _DP_DeepTensorCompute(DP_DeepTensor *dt, const int natom, const FPTYPE *coord, const int *atype, const FPTYPE *cell, FPTYPE *global_tensor, FPTYPE *force, FPTYPE *virial, FPTYPE **atomic_energy, FPTYPE *atomic_virial, int *size_at)