【问题标题】:calling a cxxfunction itself recursion调用 cxxfunction 本身递归
【发布时间】:2017-10-24 13:54:22
【问题描述】:

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

'Matmult2' 未在此范围内声明

问题也是在 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 );

')

`

【问题讨论】:

  • 你真的应该只定义具有相当短和简单源代码的内联 Rcpp 函数。我建议您使用单独的 C++ 源文件。这并不难处理(特别是如果您使用 IDE),并且您可以获得语法突出显示和其他工具的额外优势。
  • 好建议,也建议从 inline 切换到 Rcpp Attributes。

标签: r recursion inline rcpp strassen


【解决方案1】:

你不能直接递归,因为我们插入的胶水代码创建了一个被调用的包装函数——但是你不希望包装函数被递归调用,你希望你的函数叫做。所以你也需要:一个从 R 调用的包装器和一个仅限 C(++) 的递归函数 从你的包装器调用,它调用自己。

如果您切换到 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,因为后者更早溢出。)

【讨论】:

    猜你喜欢
    • 2013-09-09
    • 1970-01-01
    • 1970-01-01
    • 2020-07-16
    • 2017-11-19
    • 1970-01-01
    • 2016-03-21
    • 2011-03-27
    • 2016-01-03
    相关资源
    最近更新 更多