C++ (Cpp) Net::GetWeightsの例

コード例 #1

ファイル: genweights_mex.cpp プロジェクト: 1165048017/ConvNet

void mexFunction(int nLhs, mxArray* pLhs[], int nRhs, const mxArray* pRhs[]) {

  mexAssert(nRhs == NARGIN, "Number of input arguments in wrong!");
  mexAssert(nLhs == NARGOUT, "Number of output arguments is wrong!" );
  
  Net net;
  net.InitLayers(IN_L);
  net.InitParams(IN_P);
  net.InitWeights(NULL);
  net.GetWeights(OUT_W);
}

コード例 #2

ファイル: genweights_mex.cpp プロジェクト: yuanpu/ConvNet

void mexFunction(int nLhs, mxArray* pLhs[], int nRhs, const mxArray* pRhs[]) {

  mexAssert(nRhs == NARGIN, "Number of input arguments in wrong!");
  mexAssert(nLhs == NARGOUT, "Number of output arguments is wrong!" );
  
  size_t seed = (size_t) mexGetScalar(IN_S);
  
  Net net;
  net.InitRand(seed);
  net.InitLayers(IN_L);  
  net.InitWeights(NULL);
  net.GetWeights(OUT_W);
}