c++ - 如何计算 Caffe 中的绝对 Dtype 值？

Question

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

非常感谢您的帮助。

Answer 1

在浏览了 caffe 代码后，我找到了解决方案的答案。我可以简单地做

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