【发布时间】:2020-03-06 02:06:54
【问题描述】:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
enum Op{ADD, SUB, MUL, DIV, MATMUL};
template <typename dtype>
using AlgoFunction = double(*)(const vector<dtype> &, Op);
// for example, the sum function doesn't require template.
// just write sum(a), not sum<float>(a)
template <typename dtype>
double sum(vector<dtype> inputs) {
dtype summer = inputs[0];
for (int i=1; i<inputs.size(); i++) summer = summer + inputs[i];
return double(summer);
}
// i need to do ask this question because I perform the same
// algorithm (linearAlgo, ...) on different types of data
// (dtype = float, double, matrix<float>, matrix<double>, ...
template <typename dtype>
inline dtype numOperate(const dtype &a, const dtype &b, Op op) {
if (op==ADD) return a + b;
if (op==SUB) return a - b;
if (op==MUL) return a * b;
if (op==DIV) return a / b;
}
template <typename dtype>
double linearAlgo(const vector<dtype> &inputs, Op op) {
dtype summer = inputs[0];
for (int i=1; i<inputs.size(); i++) summer = numOperate(summer, inputs[i], op);
return double(summer);
}
template <typename dtype>
double reverseLinearAlgo(const vector<dtype> &inputs, Op op) {
int n = inputs.size();
dtype summer = inputs[n-1];
for (int i=n-2; i>=0; i--) summer = numOperate(summer, inputs[i], op);
return double(summer);
}
template<typename dtype>
vector<double> run(vector<dtype> inputs, Op op, double (*func)(const vector<dtype>&, Op)) {
vector<double> res;
res.push_back(func(inputs, op));
return res;
}
int main()
{
vector<float> a;
vector<double> b;
a.push_back(1); a.push_back(2); a.push_back(3);
b.push_back(1); b.push_back(2); b.push_back(3);
vector<double> res = run(a, ADD, linearAlgo); // allowed without specifying template
vector<double> resf = run(b, ADD, linearAlgo); // still work with multiple data type
// I want to do this assignment without specifying the template.
// in the above linear, linearAlgo (no specifying template) is possible, why not here ?
AlgoFunction<float> functor = reverseLinearAlgo; // works, but I don't want it
//AlgoFunction functor = reverseLinearAlgo; // I want to do this. compile error
vector<double> res2 = run(a, ADD, functor);
cout << res[0] << "\n";
cout << res2[0];
return 0;
}
所以我有一个函数模板指针
template <typename dtype>
using AlgoFunction = double(*)(const vector<dtype> &, Op);
指向这样的函数
template <typename dtype>
double linearAlgo(const vector<dtype> &inputs, Op op) {
dtype summer = inputs[0];
for (int i=1; i<inputs.size(); i++) summer = numOperate(summer, inputs[i], op);
return double(summer);
}
我知道在不指定模板的情况下使用模板函数指针是可能的。例如:
vector<float> a;
a.push_back(1); a.push_back(2); a.push_back(3);
vector<double> res = run(a, ADD, linearAlgo); // allowed without specifying template
但是如果我声明一个AlgoFunction 类型的变量,编译器会强制我指定模板。
//AlgoFunction<float> functor = reverseLinearAlgo; // works, but I don't want it
AlgoFunction functor = reverseLinearAlgo; // I want to do this. compile error
这不好,因为我有很多类型的数据dtype,我不想为每一个都重新指定模板。
那么我怎样才能声明AlgoFunction functor; 而不是AlgoFunction<some_datatype_name> functor; 呢?
谢谢。
编辑:目标是使用vector<AlgoFunction> functors 而不是vector<AlgoFunction<data_type> >。由于示例中res和resf都可以在不指定第三个参数的模板的情况下计算,所以我想知道vector<AlgoFunction>是否可能。
【问题讨论】:
-
“所以我有一个函数模板指针” - 或者你有别名吗?
AlgoFunction不是实例 - 它没有指向。 -
"编译器强制我指定模板" - 是的,您将
AlgoFunction设为template <typename dtype> double(*)(const vector<dtype> &, Op);的方便别名 - 编译器想知道特定的实例你喜欢。 -
"那么我如何声明
AlgoFunction functor; 而不是<AlgoFunction<some_datatype_name> functor;?" - 远远超出我的范围。据我所知,类/函数模板不会以实例的形式出现,除非被要求这样做。 -
@TedLyngmo 是的,我确实使用了别名。但是您可以在示例中看到,在将函数作为参数传递时我不需要指定模板以使其工作。所以我正在寻找一种在任何地方都可以做到这一点的方法。
-
请注意,如果您不小心,生成的代码可能会慢得多。例如,指向函数的指针可能比内联函数慢。考虑像@987654340@这样的STL算法也可能很有用。
标签: c++ function templates function-pointers function-templates