使用C++ mex获取matlab输入参数函数

Question

double learning_rate = 1; 
int training_epochs = 1; 
int k = 1; 

int train_S = 6; 
int test_S = 6; 
int visible_E = 6; 
int hidden_E = 6; 

// training data 
int train_X[6][6] = { 
    {1, 1, 1, 0, 0, 0}, 
    {1, 0, 1, 0, 0, 0}, 
    {1, 1, 1, 0, 0, 0}, 
    {0, 0, 1, 1, 1, 0}, 
    {0, 0, 1, 1, 1, 0}, 
    {0, 0, 1, 1, 1, 0} 
}; 

上面的代码是我给函数的输入参数。但我想用mexFunction将它们转换成函数，并简单地调用它们。 matlab方面有以下内容

clear * 
close all 
clc 

%% Load the data 


X= [ 1, 1, 1, 0, 0, 0; ... 
     1, 0, 1, 0, 0, 0; ... 
     1, 1, 1, 0, 0, 0; ... 
     0, 0, 1, 1, 1, 0; ... 
     0, 0, 1, 1, 1, 0; ... 
     0, 0, 1, 1, 1, 0]; 

%% Define Parameters 

numHiddenUnits = 6; 
numIterations = 1000; 
kCD = 1; 

%% Compute the RBM 

x = RBM(X, numHiddenUnits, numIterations, kCD); 

Answer 1

标量输入参数是相当简单的。矩阵输入有点棘手，因为它们使用古老的Fortran列主要命令，您可能需要在将数据发送到您的函数之前转置数据。下面是类的一个实例，你必须填补空白：

/*========================================================= 
* Built on: 
* matrixDivide.c - Example for illustrating how to use 
* LAPACK within a C MEX-file. 
* 
* This is a MEX-file for MATLAB. 
* Copyright 2009 The MathWorks, Inc. 
*=======================================================*/ 
/* $Revision: 1.1.6.2 $ $Date: 2009/05/18 19:50:18 $ */ 

#include "mex.h" 

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) 
{ 
    double * pX, * pNumHiddenUnits, * pNumIter, * pkCD; /* pointers to inputs */ 
    double * pOutput; /* output arugments */ 
    mwSignedIndex m,n;  /* matrix dimensions */ 
    int i, j; 

     /* Check for proper number of arguments. */ 
    if (nrhs != 4) 
    { 
     mexErrMsgIdAndTxt("MATLAB:RBM:rhs", 
      "This function requires 4 inputs."); 
    } 

    pX = mxGetPr(prhs[0]); /* pointer to first input, X matrix */ 
    pNumHiddenUnits = mxGetPr(prhs[1]); /* pointer to second input, scalar hidden units */ 
    pNumIter = mxGetPr(prhs[2]); /* pointer to third input, scalar number of iterations */ 
    pkCD = mxGetPr(prhs[3]); /* pointer to third input, scalar kCD */ 

    /* dimensions of input matrix */ 
    m = (mwSignedIndex)mxGetM(prhs[0]); 
    n = (mwSignedIndex)mxGetN(prhs[0]); 

    /* Validate input arguments */ 
    if (m < 1 && n < 1) 
    { 
     mexErrMsgIdAndTxt("MATLAB:RBM:notamatrix", 
      "X must be a matrix."); 
    } 

    plhs[0] = mxCreateDoubleMatrix(m, n, mxREAL); 
    pOutput = mxGetPr(plhs[0]); 

    for (i = 0; i < n; ++i) 
    { 
     for (j = 0; j < m; ++j) 
     { 
     int index = j * n + i; 
     pOutput[index] = pX[i * m + j]; 
     } 
    } 
    } 
    /* */