如果添加了print语句,则相同的Rcpp函数返回不同的输出

时间:2016-10-19 20:10:59

标签: c++ r rcpp

我使用Rcpp编写的C ++函数根据代码中是否有RcoutRprintf语句提供不同的输出。对于带有print语句1的函数,下面的代码返回H_sigma_1(),这是正确的值。但是,对于H_sigma_2(),没有print语句的函数,函数返回2.我已经在Ubuntu 16.04.1以及CentOS 6.8上对此进行了测试。虽然,我无法在Windows 10上重现此错误。因此,这似乎是一个Linux问题。

Rcpp代码

library(Rcpp)

cppFunction ( 
  "double H_sigma_1(IntegerVector sigma, NumericMatrix J, NumericVector h)
  {
    double first_sum, second_sum = 0;
    int n = sigma.size();

    for(int i = 0; i < n; i++)
    {
      for(int j = 0; j < n; j++)
      {
        // skip inside loop if i >= j to stop double counting
        if(i >= j) {continue;}
        first_sum += J(i, j) * sigma[i] * sigma[j];
        Rcout << first_sum << std::endl;
      }
      second_sum += h[i] * sigma[i];
    }
    return(-1.0 * first_sum - second_sum);
  }"
)

cppFunction (
  "double H_sigma_2(IntegerVector sigma, NumericMatrix J, NumericVector h)
  {
    double first_sum, second_sum = 0;
    int n = sigma.size();

    for(int i = 0; i < n; i++)
    {
      for(int j = 0; j < n; j++)
      {
        // skip inside loop if i >= j to stop double counting
        if(i >= j) {continue;}
        first_sum += J(i, j) * sigma[i] * sigma[j];
        // Rcout << first_sum << std::endl;
      }
      second_sum += h[i] * sigma[i];
    }
    return(-1.0 * first_sum - second_sum);
  }"
)

示例调用

设定:

n = 2
params = rep(1, n) 
h = rep(params[1], n)
J = toeplitz(c(0, params[2], rep(0, n - 2)))

测试1:

H_sigma_1(c(-1, -1), J, h)

输出:

1
[1] 1

测试2

H_sigma_2(c(-1, -1), J, h)

输出

[1] 2

1 个答案:

答案 0 :(得分:5)

您遇到的问题是在使用它们之前没有明确声明sum的起始值。

double first_sum, second_sum = 0;

与:

不同
double first_sum = 0, second_sum = 0;

请参阅编译器警告标志:

file11d36f564b15.cpp:17:5: warning: variable 'first_sum' is uninitialized when used here [-Wuninitialized]
    first_sum += J(i, j) * sigma[i] * sigma[j];
    ^~~~~~~~~
file11d36f564b15.cpp:8:21: note: initialize the variable 'first_sum' to silence this warning
    double first_sum, second_sum = 0;

您立即使用:

first_sum += J(i, j) * sigma[i] * sigma[j];

未设置first_sum = ...;

此外,另一个问题是:

second_sum = 0;

使用整数值初始化double。虽然此问题的范围很小,但要解决此问题,所有人都必须使用0.0代替0

second_sum = 0.0;

这也适用于first_sum

以上修正的代码:

library(Rcpp)

cppFunction ( 
    "double H_sigma_1(IntegerVector sigma, NumericMatrix J, NumericVector h)
    {
    double first_sum = 0.0, second_sum = 0.0;
    int n = sigma.size();

    for(int i = 0; i < n; i++) {
      for(int j = 0; j < n; j++) {
        // skip inside loop if i >= j to stop double counting
        if(i >= j) {continue;}
        first_sum += J(i, j) * sigma[i] * sigma[j];
        Rcout << first_sum << std::endl;
      }
      second_sum += h[i] * sigma[i];
    }
    return(-1.0 * first_sum - second_sum);
    }"
)

cppFunction ( 
    "double H_sigma_2(IntegerVector sigma, NumericMatrix J, NumericVector h)
    {
    double first_sum = 0.0, second_sum = 0.0;
    int n = sigma.size();

    for(int i = 0; i < n; i++) {
      for(int j = 0; j < n; j++) {
        // skip inside loop if i >= j to stop double counting
        if(i >= j) {continue;}
        first_sum += J(i, j) * sigma[i] * sigma[j];
      }
      second_sum += h[i] * sigma[i];
    }
    return(-1.0 * first_sum - second_sum);
    }"
)

测试:

n = 2
params = rep(1, n) 
h = rep(params[1], n)
J = toeplitz(c(0, params[2], rep(0, n - 2)))

H_sigma_1(c(-1, -1), J, h)

输出

1
[1] 1

测试2:

H_sigma_2(c(-1, -1), J, h)

<强>输出:

[1] 1