Skip to content
wangzhaode edited this page Feb 16, 2023 · 1 revision

Module

class Module

成员函数


Tensor

构造函数

Module() == default;

创建一个空Module

参数:无

返回:Module对象


~Module

析构函数

virtual ~Module() == default;

onForward

virtual std::vector<Express::VARP> onForward(const std::vector<Express::VARP>& inputs) = 0;

模块前向传播,返回多个结果变量

参数:

  • inputs 前向传播输入变量

返回:前向传播输出变量


forward

Express::VARP forward(Express::VARP input);

模块前向传播,返回一个结果变量

参数:

  • input 前向传播输入变量

返回:前向传播输出变量


parameters

std::vector<Express::VARP> parameters() const;

获取Module的参数

参数:无

返回:Module的参数


loadParameters

bool loadParameters(const std::vector<Express::VARP>& parameters);

加载现有的参数

参数:

  • parameters 参数值

返回:是否成功加载参数


setIsTraining

void setIsTraining(const bool isTraining);

设置Module的训练状态

参数:

  • isTraining 是否为训练模式

返回:void


getIsTraining

bool getIsTraining();

获取_Module的是否为训练模式

参数:无

返回:Module是否为训练模式,是则返回true,不是返回false


clearCache

void clearCache();

清除Module的缓存,并递归清除子模块的缓存

参数:无

返回:void


name

const std::string& name() const {
    return mName;
};

获取Module的名称

参数:无

返回:Module的名称


setName

void setName(std::string name) {
    mName = std::move(name);
};

设置Module的名称

参数:

  • name 模块的名称

返回:void


type

const std::string type() const {
    return mType;
};

获取Module的类型

参数:无

返回:Module的类型


setType

void setType(std::string type) {
    mType = std::move(type);
};

设置Module的类型

参数:

  • type 模块的类型

返回:void


addParameter

int addParameter(Express::VARP parameter);

添加参数

参数:

  • parameter 参数值

返回:添加前的参数数量


setParameter

void setParameter(Express::VARP parameter, int index);

设置参数

参数:

  • type 参数值
  • index 参数的位置索引

返回:void


createEmpty

static Module* createEmpty(const std::vector<Express::VARP>& parameters);

根据参数创建一个空的Module对象

参数:

  • parameters 参数值

返回:创建的空的Module对象


load

static Module* load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, const uint8_t* buffer, size_t length, const Config* config = nullptr);

加载module对象

参数:

  • inputs module输入信息
  • outputs module输出信息
  • buffer 缓冲信息
  • length 信息长度
  • config 其他配置项

返回:module对象


load

static Module* load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, const char* fileName, const Config* config = nullptr);

加载module对象

参数:

  • inputs module输入信息
  • outputs module输出信息
  • fileName 文件名
  • config 其他配置项

返回:module对象


load

static Module* load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, const char* fileName, const std::shared_ptr<MNN::Express::Executor::RuntimeManager> rtMgr, const Config* config = nullptr);

加载module对象

参数:

  • inputs module输入信息
  • outputs module输出信息
  • fileName 文件名
  • rtMgr 运行时资源
  • config 其他配置项

返回:module对象


load

static Module* load(const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, const uint8_t* buffer, size_t length, const std::shared_ptr<MNN::Express::Executor::RuntimeManager> rtMgr, const Config* config = nullptr);

加载module对象

参数:

  • inputs module输入信息
  • outputs module输出信息
  • buffer 缓冲信息
  • length 信息长度
  • rtMgr 运行时资源
  • config 其他配置项

返回:module对象


load

static Module* extract(std::vector<Express::VARP> inputs, std::vector<Express::VARP> outputs, bool fortrain, const std::map<std::string, SubGraph>& subGraph = {});

加载module对象

参数:

  • inputs module输入信息
  • outputs module输出信息
  • fortrain
  • subGraph 子图

返回:module对象


clone

static Module* clone(const Module* module, const bool shareParams = false);

克隆Module对象

参数:

  • module module对象实例
  • shareParams 是否共享参数,默认为false

返回:Module对象实例


getInfo

const Info* getInfo() const;

获取Module的信息

参数:无

返回:Module的信息


CloneContext

CloneContext() = default;

克隆Module的内容

参数:无

返回:Module的内容


CloneContext

explicit CloneContext(const bool shareParams)
        : mShareParams(shareParams) {};

克隆Module的内容

参数:

  • shareParams 是否共享参数

返回:Module的内容


~CloneContext

析构函数

virtual ~CloneContext() = default;

shareParams

const bool shareParams() const { return mShareParams; };

是否共享参数

参数:无

返回:共享返回true,反之则为false


getOrClone

EXPRP getOrClone(const EXPRP expr);

获取克隆的EXPRP对象

参数:

  • expr EXPRP对象值

返回:EXPRP对象


getOrClone

VARP getOrClone(const VARP var);

获取克隆的VARP对象

参数:

  • expr VARP对象值

返回:VARP对象


clone

virtual Module* clone(CloneContext* ctx) const {
    return nullptr;
};

克隆Module对象

参数:

  • ctx 克隆的上下文

返回:Module对象


registerModel

void registerModel(const std::vector<std::shared_ptr<Module>>& children);

注册子模块

参数:

  • children 子模块列表

返回:void


destroy

static void destroy(Module* m);

销毁Module对象

参数:

  • m Module对象

返回:void

Clone this wiki locally