35 lines
1.2 KiB
C
35 lines
1.2 KiB
C
#include <R.h>
|
|
#include <Rinternals.h>
|
|
#include <R_ext/Rdynload.h>
|
|
|
|
// extern SEXP FastPOI_C_sub(SEXP in_B, SEXP in_Delta,
|
|
// SEXP in_lambda, SEXP in_maxit, SEXP in_tol
|
|
// );
|
|
|
|
/* Tensor Times Matrix a.k.a. Mode Product */
|
|
extern SEXP ttm(SEXP A, SEXP X, SEXP mode, SEXP op);
|
|
|
|
/* Matrix Times Vectorized Kronecker product `A vec(B_1 %x% ... %x% B_r)` */
|
|
extern SEXP mtvk(SEXP A, SEXP Bs);
|
|
|
|
/* Tensor Mode Crossproduct `A_(m) B_(m)^T` */
|
|
extern SEXP mcrossprod(SEXP A, SEXP B, SEXP mode);
|
|
/* Symmetric Tensor Mode Crossproduct `A_(m) A_(m)^T` */
|
|
extern SEXP mcrossprod_sym(SEXP A, SEXP mode);
|
|
|
|
/* List of registered routines (a.k.a. C entry points) */
|
|
static const R_CallMethodDef CallEntries[] = {
|
|
// {"FastPOI_C_sub", (DL_FUNC) &FastPOI_C_sub, 5}, // NOT USED
|
|
{"C_ttm", (DL_FUNC) &ttm, 4},
|
|
{"C_mtvk", (DL_FUNC) &mtvk, 2},
|
|
{"C_mcrossprod", (DL_FUNC) &mcrossprod, 3},
|
|
{"C_mcrossprod_sym", (DL_FUNC) &mcrossprod_sym, 2},
|
|
{NULL, NULL, 0}
|
|
};
|
|
|
|
/* Restrict C entry points to registered routines. */
|
|
void R_init_tensorPredictors(DllInfo *dll) {
|
|
R_registerRoutines(dll, NULL, CallEntries, NULL, NULL);
|
|
R_useDynamicSymbols(dll, FALSE);
|
|
}
|