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.

  • out[out] The output vector.

  • in[in] The input vector.

  • fwd_map[in] The map.

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