调用 cxxfunction 本身递归

Question

我在 R 中的 cxx 函数有问题。我想自己调用它，不幸的是，编译器给我这个错误消息：

'Matmult2' is not declared in this scope

问题也是在 cxx 函数中调用 Matmult2。

我原来的问题与strassen算法有关，我想用RCPP / Inline递归调用。非常感谢您的帮助！

`

Matmult2<- cxxfunction(signature( a="NumericMatrix",b="NumericMatrix"), plugin="RcppArmadillo",includes = rcpp_inc,
  body='

              Rcpp::NumericMatrix verg(a);
              int n = verg.nrow();

              arma::mat A = Rcpp::as< arma::mat >(a);
              arma::mat B = Rcpp::as< arma::mat >(b);

              arma::mat a11 = A(span(0,(n/2)-1), span(0,(n/2)-1));
              arma::mat a12 = A(span(0,(n/2)-1), span((n/2),n-1));
              arma::mat a21 = A(span((n/2),n-1), span(0,(n/2)-1));
              arma::mat a22 = A(span((n/2),n-1), span((n/2),n-1));

              arma::mat b11 = B(span(0,(n/2)-1), span(0,(n/2)-1));
              arma::mat b12 = B(span(0,(n/2)-1), span((n/2),n-1));
              arma::mat b21 = B(span((n/2),n-1), span(0,(n/2)-1));
              arma::mat b22 = B(span((n/2),n-1), span((n/2),n-1));


              if ( n  < 780) return Rcpp::wrap( A * B );


              arma::mat P1(Matmult2(a11,(b12-b22)));
              arma::mat P2(Matmult2((a11+a12),(b22)));
              arma::mat P3(Matmult2((a21+a22),(b11)));
              arma::mat P4(Matmult2((a22),(b21-b11)));
              arma::mat P5(Matmult2((a11+a22),(b11+b22)));
              arma::mat P6(Matmult2((a12-a22),(b21+b22)));
              arma::mat P7(Matmult2((a11-a21),(b11+b12)));

              arma::mat c11((P5+P4-P2+P6));
              arma::mat c12((P1+P2));
              arma::mat c21((P3+P4));
              arma::mat c22((P5+P1-P3-P7));

              mat C1 = join_rows(c11, c12);
              mat C2 = join_rows(c21, c22);
              mat C = join_cols(C1, C2);

              return Rcpp::wrap( C );

')

`

Answer 1

你不能直接递归，因为我们插入的胶水代码创建了一个被调用的包装函数---但你不希望包装函数被递归调用，您希望调用您的函数。所以你也需要：一个从 R 调用的包装器，和一个 C(++)-only 递归函数，它是从您的包装器调用，并调用自身。

如果您切换到 Rcpp 属性，它也可以工作。这里我们使用递归函数的旗手，斐波那契数列：

R> library(Rcpp)
R> cppFunction("double fib(double n) { if (n < 2) return n; return(fib(n-1) + fib(n-2)); }")
R> sapply(0:10, fib)
 [1]  0  1  1  2  3  5  8 13 21 34 55
R>

（我使用 double 而不是 int，因为后者更早溢出。）

调用 cxxfunction 本身递归

calling a cxxfunction itself recursion

recursion

r

inline

strassen

rcpp