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.