Template Function deepmd::session_input_tensors(std::vector<std::pair<std::string, tensorflow::Tensor>>&, const std::vector<VALUETYPE>&, const int&, const std::vector<int>&, const std::vector<VALUETYPE>&, InputNlist&, const std::vector<VALUETYPE>&, const std::vector<VALUETYPE>&, const deepmd::AtomMap&, const int, const int, const std::string, const bool)
Defined in File common.h
Function Documentation
- template<typename MODELTYPE, typename VALUETYPE>
int deepmd::session_input_tensors(std::vector<std::pair<std::string, tensorflow::Tensor>> &input_tensors, const std::vector<VALUETYPE> &dcoord_, const int &ntypes, const std::vector<int> &datype_, const std::vector<VALUETYPE> &dbox, InputNlist &dlist, const std::vector<VALUETYPE> &fparam_, const std::vector<VALUETYPE> &aparam_, const deepmd::AtomMap &atommap, const int nghost, const int ago, const std::string scope = "", const bool aparam_nall = false) Get input tensors.
- Parameters
input_tensors – [out] Input tensors.
dcoord_ – [in] Coordinates of atoms.
ntypes – [in] Number of atom types.
datype_ – [in] Atom types.
dlist – [in] Neighbor list.
fparam_ – [in] Frame parameters.
aparam_ – [in] Atom parameters.
atommap – [in] Atom map.
nghost – [in] Number of ghost atoms.
ago – [in] Update the internal neighbour list if ago is 0.
scope – [in] The scope of the tensors.
aparam_nall – [in] Whether the atomic dimesion of atomic parameters is nall.