コード例 #1
0
int luaopen_libnn(lua_State *L)
{
  lua_newtable(L);
  lua_pushvalue(L, -1);
  lua_setglobal(L, "nn");

  nn_FloatAbsMultiMarginCriterion_init(L);
  nn_FloatSqrt_init(L);
  nn_FloatSquare_init(L);
  nn_FloatHardTanh_init(L);
  nn_FloatLogSoftMax_init(L);
  nn_FloatClassNLLCriterion_init(L);
  nn_FloatMSECriterion_init(L);
  nn_FloatSmoothL1Criterion_init(L);
  nn_FloatMarginCriterion_init(L);
  nn_FloatDistKLDivCriterion_init(L);
  nn_FloatLogSigmoid_init(L);
  nn_FloatSigmoid_init(L);
  nn_FloatSoftMax_init(L);
  nn_FloatSoftPlus_init(L);
  nn_FloatTanh_init(L);
  nn_FloatHardShrink_init(L);
  nn_FloatSoftShrink_init(L);
  nn_FloatThreshold_init(L);
  nn_FloatPReLU_init(L);
  nn_FloatRReLU_init(L);
  nn_FloatELU_init(L);
  nn_FloatLeakyReLU_init(L);
  nn_FloatSparseLinear_init(L);
  nn_FloatTemporalConvolution_init(L);
  nn_FloatTemporalSubSampling_init(L);
  nn_FloatTemporalMaxPooling_init(L);
  nn_FloatSpatialConvolution_init(L);
  nn_FloatSpatialFullConvolution_init(L);
  nn_FloatSpatialFullConvolutionMap_init(L);
  nn_FloatSpatialConvolutionMM_init(L);
  nn_FloatSpatialConvolutionMap_init(L);
  nn_FloatSpatialSubSampling_init(L);
  nn_FloatSpatialMaxPooling_init(L);
  nn_FloatSpatialMaxUnpooling_init(L);
  nn_FloatSpatialFractionalMaxPooling_init(L);
  nn_FloatSpatialAveragePooling_init(L);
  nn_FloatSpatialAdaptiveMaxPooling_init(L);
  nn_FloatVolumetricConvolution_init(L);
  nn_FloatVolumetricConvolutionMM_init(L);
  nn_FloatVolumetricFullConvolution_init(L);
  nn_FloatVolumetricMaxPooling_init(L);
  nn_FloatVolumetricAveragePooling_init(L);
  nn_FloatMultiMarginCriterion_init(L);
  nn_FloatMultiLabelMarginCriterion_init(L);
  nn_FloatL1Cost_init(L);
  nn_FloatSpatialUpSamplingNearest_init(L);
  nn_FloatLookupTable_init(L);

  nn_DoubleAbsMultiMarginCriterion_init(L);
  nn_DoubleSqrt_init(L);
  nn_DoubleSquare_init(L);
  nn_DoubleHardTanh_init(L);
  nn_DoubleLogSoftMax_init(L);
  nn_DoubleClassNLLCriterion_init(L);
  nn_DoubleMSECriterion_init(L);
  nn_DoubleSmoothL1Criterion_init(L);
  nn_DoubleMarginCriterion_init(L);
  nn_DoubleDistKLDivCriterion_init(L);
  nn_DoubleLogSigmoid_init(L);
  nn_DoubleSigmoid_init(L);
  nn_DoubleSoftMax_init(L);
  nn_DoubleSoftPlus_init(L);
  nn_DoubleTanh_init(L);
  nn_DoubleHardShrink_init(L);
  nn_DoubleSoftShrink_init(L);
  nn_DoubleThreshold_init(L);
  nn_DoublePReLU_init(L);
  nn_DoubleRReLU_init(L);
  nn_DoubleELU_init(L);
  nn_DoubleLeakyReLU_init(L);
  nn_DoubleSparseLinear_init(L);
  nn_DoubleTemporalConvolution_init(L);
  nn_DoubleTemporalSubSampling_init(L);
  nn_DoubleTemporalMaxPooling_init(L);
  nn_DoubleSpatialMaxUnpooling_init(L);
  nn_DoubleSpatialConvolution_init(L);
  nn_DoubleSpatialFullConvolution_init(L);
  nn_DoubleSpatialFullConvolutionMap_init(L);
  nn_DoubleSpatialConvolutionMM_init(L);
  nn_DoubleSpatialFullConvolution_init(L);
  nn_DoubleSpatialConvolutionMap_init(L);
  nn_DoubleSpatialSubSampling_init(L);
  nn_DoubleSpatialMaxPooling_init(L);
  nn_DoubleSpatialFractionalMaxPooling_init(L);
  nn_DoubleSpatialAveragePooling_init(L);
  nn_DoubleSpatialAdaptiveMaxPooling_init(L);
  nn_DoubleVolumetricConvolution_init(L);
  nn_DoubleVolumetricConvolutionMM_init(L);
  nn_DoubleVolumetricFullConvolution_init(L);
  nn_DoubleVolumetricMaxPooling_init(L);
  nn_DoubleVolumetricAveragePooling_init(L);
  nn_DoubleMultiMarginCriterion_init(L);
  nn_DoubleMultiLabelMarginCriterion_init(L);
  nn_DoubleL1Cost_init(L);
  nn_DoubleSpatialUpSamplingNearest_init(L);
  nn_DoubleLookupTable_init(L);

  return 1;
}
コード例 #2
0
ファイル: init.c プロジェクト: aironashish/nn
int luaopen_libnn(lua_State *L)
{
  lua_newtable(L);
  lua_pushvalue(L, -1);
  lua_setglobal(L, "nn");

  nn_FloatSqrt_init(L);
  nn_FloatSquare_init(L);
  nn_FloatMSECriterion_init(L);
  nn_FloatSmoothL1Criterion_init(L);
  nn_FloatMarginCriterion_init(L);
  nn_FloatSigmoid_init(L);
  nn_FloatSoftMax_init(L);
  nn_FloatSoftPlus_init(L);
  nn_FloatTanh_init(L);
  nn_FloatSoftShrink_init(L);
  nn_FloatThreshold_init(L);
  nn_FloatPReLU_init(L);
  nn_FloatRReLU_init(L);
  nn_FloatSparseLinear_init(L);
  nn_FloatTemporalConvolution_init(L);
  nn_FloatTemporalSubSampling_init(L);
  nn_FloatTemporalMaxPooling_init(L);
  nn_FloatSpatialBatchNormalization_init(L);
  nn_FloatSpatialConvolution_init(L);
  nn_FloatSpatialConvolutionLocal_init(L);
  nn_FloatSpatialFullConvolution_init(L);
  nn_FloatSpatialFullConvolutionMap_init(L);
  nn_FloatSpatialConvolutionMM_init(L);
  nn_FloatSpatialConvolutionMap_init(L);
  nn_FloatSpatialSubSampling_init(L);
  nn_FloatSpatialMaxPooling_init(L);
  nn_FloatSpatialMaxUnpooling_init(L);
  nn_FloatSpatialFractionalMaxPooling_init(L);
  nn_FloatSpatialAveragePooling_init(L);
  nn_FloatSpatialAdaptiveMaxPooling_init(L);
  nn_FloatVolumetricConvolution_init(L);
  nn_FloatVolumetricConvolutionMM_init(L);
  nn_FloatVolumetricFullConvolution_init(L);
  nn_FloatVolumetricMaxPooling_init(L);
  nn_FloatVolumetricMaxUnpooling_init(L);
  nn_FloatVolumetricAveragePooling_init(L);
  nn_FloatMultiMarginCriterion_init(L);
  nn_FloatMultiLabelMarginCriterion_init(L);
  nn_FloatSpatialUpSamplingNearest_init(L);

  nn_DoubleSqrt_init(L);
  nn_DoubleSquare_init(L);
  nn_DoubleMSECriterion_init(L);
  nn_DoubleSmoothL1Criterion_init(L);
  nn_DoubleMarginCriterion_init(L);
  nn_DoubleSigmoid_init(L);
  nn_DoubleSoftMax_init(L);
  nn_DoubleSoftPlus_init(L);
  nn_DoubleTanh_init(L);
  nn_DoubleSoftShrink_init(L);
  nn_DoubleThreshold_init(L);
  nn_DoublePReLU_init(L);
  nn_DoubleRReLU_init(L);
  nn_DoubleSparseLinear_init(L);
  nn_DoubleTemporalConvolution_init(L);
  nn_DoubleTemporalSubSampling_init(L);
  nn_DoubleTemporalMaxPooling_init(L);
  nn_DoubleSpatialBatchNormalization_init(L);
  nn_DoubleSpatialMaxUnpooling_init(L);
  nn_DoubleSpatialConvolution_init(L);
  nn_DoubleSpatialConvolutionLocal_init(L);
  nn_DoubleSpatialFullConvolution_init(L);
  nn_DoubleSpatialFullConvolutionMap_init(L);
  nn_DoubleSpatialConvolutionMM_init(L);
  nn_DoubleSpatialFullConvolution_init(L);
  nn_DoubleSpatialConvolutionMap_init(L);
  nn_DoubleSpatialSubSampling_init(L);
  nn_DoubleSpatialMaxPooling_init(L);
  nn_DoubleSpatialFractionalMaxPooling_init(L);
  nn_DoubleSpatialAveragePooling_init(L);
  nn_DoubleSpatialAdaptiveMaxPooling_init(L);
  nn_DoubleVolumetricConvolution_init(L);
  nn_DoubleVolumetricConvolutionMM_init(L);
  nn_DoubleVolumetricFullConvolution_init(L);
  nn_DoubleVolumetricMaxPooling_init(L);
  nn_DoubleVolumetricMaxUnpooling_init(L);
  nn_DoubleVolumetricAveragePooling_init(L);
  nn_DoubleMultiMarginCriterion_init(L);
  nn_DoubleMultiLabelMarginCriterion_init(L);
  nn_DoubleSpatialUpSamplingNearest_init(L);

  return 1;
}