Question

我正在尝试在Rcpp中实现一个简单的代码来计算和填充距离矩阵的条目。问题是Rcpp代码（下面）返回一个矩阵D，其中所有元素的值都为零。这个问题似乎没有在论坛的任何地方得到解决 - 我很感激一些建议！

  src_d_err_c <- '
  using namespace Rcpp;
  double d_err_c(NumericVector cx, NumericVector csx, NumericVector cy, NumericVector csy) {
    using namespace Rcpp;    
    NumericVector d = (cx - cy)*(cx - cy) / csx;
    double s = std::accumulate(d.begin(), d.end(), 0.0);

    return s;
  }'

  src_d_mat = '   
  using namespace Rcpp;
  // input

  Rcpp::NumericMatrix cX(X);
  Rcpp::NumericMatrix cY(Y);
  Rcpp::NumericMatrix cSX(SX);
  Rcpp::NumericMatrix cSY(SY);  
  int N1 = cX.nrow();
  int N2 = cY.nrow();
  NumericMatrix D(N1, N2);
  NumericVector v(N1);

  for (int x = 0; x++; x<N1){
    v[x] = x;
    for (int y = 0; y++; y<N2) {
      D(x,y) = d_err_c(cX(x,_), cSX(x,_), cY(y,_), cSY(y,_));
    };
  };
  return wrap(v);'  
  fun <- cxxfunction(signature(X = "numeric",  SX = "numeric", 
                               Y = "numeric",  SY = "numeric"),
                     body = src_d_mat, includes = src_d_err_c, 
                     plugin = "Rcpp")

Answer 1

for循环的参数顺序错误：条件应该在中间，而增量在结尾。

for (int x = 0; x < N1; x++)

Answer 2

@Vincent正确地指出了一个主要错误（实际上没有循环），但还有另一个主要：当您的计算进入v时返回D什么意思回来。（实际上你也根本不需要v。）

这是一个修复版本，它使用正确的循环索引，返回正确的对象，并省略了未使用的代码。它还切换到Rcpp属性。

将其保存在文件中，例如“distmat.cpp”，然后使用sourceCpp("distmat.cpp")，之后您可以调用新功能d_mat。

#include <Rcpp.h>

using namespace Rcpp;

double d_err_c(NumericVector cx, NumericVector csx, 
               NumericVector cy, NumericVector csy) {
    NumericVector d = (cx - cy)*(cx - cy) / csx;
    return std::accumulate(d.begin(), d.end(), 0.0);
}

// [[Rcpp::export]]
NumericMatrix d_mat(NumericMatrix cX, NumericMatrix cY, 
                    NumericMatrix cSX, NumericMatrix cSY) {
    int N1 = cX.nrow();
    int N2 = cY.nrow();
    NumericMatrix D(N1, N2);

    for (int x = 0; x<N1; x++) {
        for (int y = 0; y<N2; y++) {
            D(x,y) = d_err_c(cX(x,_), cSX(x,_), cY(y,_), cSY(y,_));
        }
    }
    return D;
}

NumericMatrix返回全零的矩阵？

2 个答案: