Template Function deepmd::select_map(std::vector<VT>&, const std::vector<VT>&, const std::vector<int>&, const int&, const int&, const int&, const int&)

Function Documentation

template<typename VT>
void deepmd::select_map(std::vector<VT> &out, const std::vector<VT> &in, const std::vector<int> &fwd_map, const int &stride, const int &nframes = 1, const int &nall1 = 0, const int &nall2 = 0)

Apply the given map to a vector.

Parameters
  • out – [out] The output vector.

  • in – [in] The input vector.

  • fwd_map – [in] The map.

  • stride – [in] The stride of the input vector.

  • nframes – [in] The number of frames.

  • nall1 – [in] The number of atoms in the input vector.

  • nall2 – [in] The number of atoms in the output vector.