Template Function deepmd::hpp::select_map

Function Documentation

template<typename VT>
void deepmd::hpp::select_map(std::vector<VT> &out, const std::vector<VT> &in, const std::vector<int> &fwd_map, const int &stride)

Apply the given map to a vector. Assume nframes is 1.

Template Parameters

VT – The value type of the vector. Only support int.

Parameters
  • out – [out] The output vector.

  • in – [in] The input vector.

  • fwd_map – [in] The map.

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