如何在Caffe中计算绝对Dtype值?

时间:2018-03-02 11:49:21

标签: c++ caffe

我正致力于在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?? 

非常感谢你的帮助。

1 个答案:

答案 0 :(得分:2)

在查看了caffe代码后,我找到了解决方案的答案。 我可以简单地做

Dtype myval = Dtype( std::abs(A+B+prob_data[idx]) );