我正致力于在Caffe中创建自定义图层。我需要为几个元素(不是整个blob)计算Dtype的绝对值。我应该使用哪种赦免功能?
const Dtype* prob_data = bottom[0]->cpu_data();
const Dtype* label = bottom[1]->cpu_data();
...
const int idx = ...
Dtype A = Dtype(-20); // example
Dtype B = Dtype(10); // example
...
Dtype myval = fabs(A+B+prob_data[idx]); // which abs function to be used here??
非常感谢你的帮助。
答案 0 :(得分:2)
在查看了caffe代码后,我找到了解决方案的答案。 我可以简单地做
Dtype myval = Dtype( std::abs(A+B+prob_data[idx]) );