(Rcpp)armadillos abs() 函数在与 c++ double 一起使用时输出假值，而 std::abs 工作

问题描述

考虑这个 R 函数：

r_abs <- function(x,y,z){
    2 * abs((x >= y) - z)
}

还有那 2 个 RcppArmadillo 等价物：

// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::depends(RcppProgress)]]
#include <RcppArmadillo.h>
#include <string>
#include <progress.hpp>
using namespace arma;

// [[Rcpp::export]]
double arma_abs(const double &x,const double &y,const double &z)
{
    return 2 * abs((x >= y) - z);
}

// [[Rcpp::export]]
double std_abs(const double &x,const double &z)
{
    return 2 * std::abs((x >= y) - z);
}

现在问题来了：

sourceCpp(".test/test.cpp")

x <- 1
y <- 2
z <- 0.5

r_abs(x,z)
[1] 1
arma_abs(x,z)
[1] 0
std_abs(x,z)
[1] 1

为什么 arma_abs() 在这里输出 0？该问题仅发生在 -1 < z < 1。

非常感谢您能提供的任何帮助。

解决方法

这是 C 和 C++ 的一个棘手且有些乏味的问题部分修复由最近添加的相关 std::abs() 修复。

如果您使用的是 Unix 系统，请尝试 man 3 abs。您对（不合格的）abs() 的使用等价于 ::abs()，它从 int abs(int) 获得 stdlib.h。

对 int 的强制转换使参数 -0.5 截断为零，乘以二仍然为零。简而言之，arma 与此无关。 （并且犰狳通常只有向量、矩阵、立方体、...类型的函数，而不是像 double 这样的原子 C/C++ 类型）。

为了完整起见，我的代码版本在下面嵌入了 R 部分。我删除了未使用的引用 RcppProgress。

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

//using namespace arma;

// [[Rcpp::export]]
double arma_abs(const double &x,const double &y,const double &z) {
  auto val = (x >= y) - z;
  Rcpp::Rcout << "val " << val << std::endl;
  Rcpp::Rcout << "int(val) " << int(val) << std::endl;
  return 2 * ::abs((x >= y) - z);
}

// [[Rcpp::export]]
double std_abs(const double &x,const double &z) {
    return 2 * std::abs((x >= y) - z);
}

/*** R

r_abs <- function(x,y,z){
    2 * abs((x >= y) - z)
}

x <- 1
y <- 2
z <- 0.5

r_abs(x,z)           # 1
arma_abs(x,z)      # 0
std_abs(x,z)       # 1
*/

编辑： 更正了错别字。我在上面写了 std::fabs() 的意思是 std::abs()。 fabs() 是我们一直必须使用的，以避免在这里看到的强制转换为 int 的问题。