MNN/docs/cpp/Expr.md

1024 lines
13 KiB
Markdown
Raw Permalink Normal View History

# VARP
```cpp
class VARP
```
## 枚举类
### Dimensionformat
```cpp
enum Dimensionformat {
NHWC,
NC4HW4,
NCHW
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `NHWC` | |
| 1 | `NC4HW4` | |
| 2 | `NCHW` | |
---
### InputType
```cpp
enum InputType {
INPUT = 0,
CONSTANT = 1,
TRAINABLE = 2,
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `INPUT` | 默认输入变量 |
| 1 | `CONSTANT` | 常量 |
| 2 | `TRAINABLE` | 可训练变量 |
## 成员函数
---
### VARP
```cpp
VARP(std::shared_ptr<Variable> c) {
mContent = std::move(c);
};
```
调用移动构造函数将c的对象移动至mContent对象
参数:
- `c` Variable类型变量
返回:`void`
---
### VARP
```cpp
VARP(Variable* c) {
mContent.reset(c);
};
```
重置mContent对象中的Variable对象
参数:
- `c` Variable类型变量
返回:`void`
---
### get
```cpp
Variable* get() const {
return mContent.get();
};
```
获取mContent的Variable对象
参数:无
返回Variable对象
---
### VARP
```cpp
VARP(const VARP& var) {
mContent = var.mContent;
};
```
把var的content对象赋值给mContent变量
参数:
- `var` 输入变量
返回:`void`
---
### VARP
```cpp
VARP(VARP&& var) {
mContent = std::move(var.mContent);
};
```
调用移动构造函数将var.mContent的对象移动至mContent对象
参数:
- `var` 输入变量
返回:`void`
---
### operator+
```cpp
VARP operator+(VARP var) const;
```
VARP类型对象加法计算
参数:
- `var` 输入变量
返回VARP对象
---
### operator-
```cpp
VARP operator-(VARP var) const;
```
VARP类型对象减法计算
参数:
- `var` 输入变量
返回VARP对象
---
### operator*
```cpp
VARP operator*(VARP var) const;
```
VARP类型对象乘法计算
参数:
- `var` 输入变量
返回VARP对象
---
### operator/
```cpp
VARP operator/(VARP var) const;
```
VARP类型对象除法计算
参数:
- `var` 输入变量
返回VARP对象
---
### mean
```cpp
VARP mean(INTS dims) const;
```
计算均值
参数:
- `dims` 一个向量
返回VARP对象
---
### sum
```cpp
VARP sum(INTS dims) const;
```
计算和
参数:
- `dims` 一个向量
返回VARP对象
---
### operator==
```cpp
bool operator==(const VARP& var) const {
return var.mContent == mContent;
};
```
重载相等运算符判断var.mContent和mContent是否相等
参数:
- `var` 输入变量
返回true/false
---
### operator<
```cpp
bool operator<(const VARP& var) const {
return mContent < var.mContent;
};
```
重载小于运算符判断mContent是否小于var.mContent
参数:
- `var` 输入变量
返回true/false
---
### operator<=
```cpp
bool operator<=(const VARP& var) const {
return mContent <= var.mContent;
};
```
重载小于等于运算符判断mContent是否小于等于var.mContent
参数:
- `var` 输入变量
返回true/false
---
### operator=
```cpp
VARP& operator=(const VARP& var) {
mContent = var.mContent;
return *this;
};
```
拷贝var.mContent对象到mContent
参数:
- `var` 输入变量
返回:当前对象的拷贝
---
### operator=
```cpp
VARP& operator=(Variable* var) {
mContent.reset(var);
return *this;
};
```
重置mContent对象中的Variable对象并拷贝
参数:
- `var` 输入变量
返回:当前对象的拷贝
---
### operator->
```cpp
Variable* operator->() const {
return mContent.get();
};
```
获取mContent对象的值
参数:无
返回Variable对象
---
### fix
```cpp
bool fix(InputType type) const;
```
朝零方向取整
参数:
- `type` 输入数据类型
返回true/false
---
### operator==
```cpp
inline bool operator==(Variable* src, VARP dst) {
return src == dst.get();
};
```
重载相等运算符判断src和dst.get()是否相等
参数:
- `src` Variable类型输入变量
- `dst` VARP类型输入变量
返回true/false
---
### operator!=
```cpp
inline bool operator!=(Variable* src, VARP dst) {
return src != dst.get();
};
```
重载不相等运算符判断src和dst.get()是否不相等
参数:
- `src` Variable类型输入变量
- `dst` VARP类型输入变量
返回true/false
# Variable
```cpp
class Variable
```
## 成员函数
---
### name
```cpp
const std::string& name() const;
```
获取Variable对象的名称
参数:无
返回:名称
---
### setName
```cpp
void setName(const std::string& name);
```
设置名称
参数:
- `name` 名称
返回:`void`
---
### expr
```cpp
std::pair<EXPRP, int> expr() const {
return std::make_pair(mFrom, mFromIndex);
};
```
创建一个EXPRP对象需要mFrom作为参数位置为mFromIndex
参数:无
返回EXPRP对象
---
### getInfo
```cpp
const Info* getInfo();
```
获取Variable对象的相关信息
参数:无
返回如果计算信息错误返回nullptr
---
### resize
```cpp
bool resize(INTS dims);
```
调整Variable对象的大小
参数:
- `dims` 一个向量
返回true/false
---
### readMap
```cpp
template <typename T>
const T* readMap() {
return (const T*)readInternal();
};
```
读取内部信息
参数:无
返回:信息
---
### writeMap
```cpp
template <typename T>
T* writeMap() {
return (T*)writeInternal();
};
```
写入内部信息
参数:无
返回:信息
---
### input
```cpp
bool input(VARP src);
```
输入信息
参数:
- `src` 输入变量
返回true/false
---
### replace
```cpp
static void replace(VARP dst, VARP src);
```
替换信息
参数:
- `dst` 输入变量
- `src` 输入变量
返回:`void`
---
### create
```cpp
static VARP create(EXPRP expr, int index = 0);
```
在index位置创建expr对象
参数:
- `expr` 输入变量
- `index` 位置下标默认为0
返回VARP对象
---
### load
```cpp
static std::vector<VARP> load(const char* fileName);
```
通过文件名加载对象
参数:
- `fileName` 文件名
返回VARP对象矩阵
---
### loadMap
```cpp
static std::map<std::string, VARP> loadMap(const char* fileName);
```
通过文件名读取模型对象
参数:
- `fileName` 文件名
返回:模型对象
---
### load
```cpp
static std::vector<VARP> load(const uint8_t* buffer, size_t length);
```
加载存储的长度为length的模型对象
参数:
- `buffer` 存储数据
- `length` 数据长度
返回:模型对象
---
### loadMap
```cpp
static std::map<std::string, VARP> loadMap(const uint8_t* buffer, size_t length);
```
读取存储的长度为length的模型对象
参数:
- `buffer` 存储数据
- `length` 数据长度
返回:模型对象
---
### getInputAndOutput
```cpp
static std::pair<std::map<std::string, VARP>, std::map<std::string, VARP>> getInputAndOutput(const std::map<std::string, VARP>& allVariable);
```
获取模型输入输出节点
参数:
- `allVariable` 输入变量
返回:模型输入输出节点
---
### mapToSequence
```cpp
static std::vector<VARP> mapToSequence(const std::map<std::string, VARP>& source);
```
模型的输出节点及其名称
参数:
- `source` 输入变量
返回:输出节点及其名称
---
### getExecuteOrder
```cpp
static std::vector<EXPRP> getExecuteOrder(const std::vector<VARP>& output);
```
获取操作指令
参数:
- `output` 输入变量
返回:指令集
---
### save
```cpp
static void save(const std::vector<VARP>& vars, const char* fileName);
```
保存vars到指定位置
参数:
- `vars` 输入变量
- `fileName` 文件名
返回:`void`
---
### save
```cpp
static std::vector<int8_t> save(const std::vector<VARP>& vars);
```
保存vars到默认位置
参数:
- `vars` 输入变量
返回:`void`
---
### save
```cpp
static void save(const std::vector<VARP>& vars, NetT* dest);
```
保存vars到dest位置
参数:
- `vars` 输入变量
- `dest` 目标地址
返回:`void`
---
### prepareCompute
```cpp
static void prepareCompute(const std::vector<VARP>& vars, bool forceCPU = false);
```
将几个变量打包在一个管道中进行计算
参数:
- `vars` 输入变量
- `forceCPU` 是否强制使用CPU默认为false
返回:`void`
---
### compute
```cpp
static void compute(const std::vector<VARP>& vars, bool forceCPU = false);
```
计算变量
参数:
- `vars` 输入变量
- `forceCPU` 是否强制使用CPU默认为false
返回:`void`
---
### linkNumber
```cpp
size_t linkNumber() const;
```
获取输出信息的size
参数:无
返回size
---
### toExprs
```cpp
const std::vector<WeakEXPRP>& toExprs() const;
```
返回模型对象信息
参数:无
返回:模型对象信息
---
### setExpr
```cpp
void setExpr(EXPRP expr, int index) {
mFrom = expr;
mFromIndex = index;
};
```
在index位置设置EXPRP对象
参数:无
返回EXPRP对象
# Expr
```cpp
class Expr
```
## 枚举类
---
### MemoryType
```cpp
enum MemoryType {
COPY,
MOVE,
REF
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `COPY` | 拷贝 |
| 1 | `MOVE` | 移动 |
| 2 | `REF` | 引用 |
## 成员函数
---
### create
```cpp
static EXPRP create(Tensor* tensor, bool own = false);
```
创建一个包含tensor的EXPRP变量
参数:
- `tensor` 输入变量
- `own` 默认为false
返回EXPRP变量
---
### create
```cpp
static EXPRP create(Variable::Info&& info, const void* ptr, VARP::InputType type, MemoryType copy = COPY);
```
创建一个包含info对象的EXPRP变量
参数:
- `info` Variable类型输入变量
- `ptr` 目标对象地址
- `type` 输入数据类型
- `copy` 内存拷贝
返回EXPRP变量
---
### create
```cpp
static EXPRP create(const OpT* op, std::vector<VARP> inputs, int outputSize = 1);
```
创建EXPRP变量
参数:
- `op` 输入变量
- `inputs` 输入变量
- `outputSize` 输出信息大小默认为1
返回EXPRP变量
---
### create
```cpp
static EXPRP create(std::shared_ptr<BufferStorage> extra, std::vector<VARP>&& inputs, int outputSize = 1);
```
创建EXPRP变量
参数:
- `extra` 输入变量
- `inputs` 输入变量
- `outputSize` 输出信息大小默认为1
返回EXPRP变量
---
### create
```cpp
static EXPRP create(std::unique_ptr<OpT>&& op, std::vector<VARP> inputs, int outputSize = 1) {
return create(op.get(), inputs, outputSize);
};
```
创建EXPRP变量
参数:
- `op` 输入变量
- `inputs` 输入变量
- `outputSize` 输出信息大小默认为1
返回EXPRP变量
---
### setName
```cpp
void setName(const std::string& name);
```
设置名称
参数:
- `name` 名称
返回:`void`
---
### get
```cpp
const Op* get() const {
return mOp;
};
```
获取对象信息
参数:无
返回:对象信息
---
### inputs
```cpp
const std::vector<VARP>& inputs() const {
return mInputs;
};
```
获取输入节点信息
参数:无
返回:输入节点信息
---
### outputSize
```cpp
int outputSize() const {
return (int)mOutputNames.size();
};
```
返回输出节点信息大小
参数:无
返回size
---
### replace
```cpp
static void replace(EXPRP oldExpr, EXPRP newExpr);
```
用newExpr替换oldExpr
参数:
- `oldExpr` 输入变量,源对象
- `newExpr` 输入变量,目标对象
返回:`void`
---
### requireInfo
```cpp
bool requireInfo();
```
获取需要的信息
参数:无
返回:信息
---
### visitOutputs
```cpp
void visitOutputs(const std::function<bool(EXPRP, int)>& visit);
```
访问输出节点的信息
参数:
- `visit` 访问方法
返回:`void`
---
### visit
```cpp
static void visit(EXPRP expr, const std::function<bool(EXPRP)>& before, const std::function<bool(EXPRP)>& after);
```
访问某一个范围的信息
参数:
- `expr` 输入变量
- `before` before指针
- `after` after指针
返回:`void`
---
### outputs
```cpp
const std::vector<WeakEXPRP>& outputs() const {
return mTo;
};
```
返回输出节点信息
参数:无
返回:信息
---
### ~Expr()
析构函数
---
### visited
```cpp
bool visited() const {
return mVisited;
};
```
是否已经访问过
参数:无
返回true/false
---
### setVisited
```cpp
void setVisited(bool visited) {
mVisited = visited;
};
```
设置已经被访问
参数:
- `visited` 是否访问
返回:`void`
---
### name
```cpp
const std::string& name() const {
return mName;
};
```
获取名称
参数:无
返回:名称
---
### outputName
```cpp
const std::string& outputName(int index) {
return mOutputNames[index];
};
```
输出指定index的名称
参数:
- `index` 下标
返回:名称
---
### inputType
```cpp
VARP::InputType inputType() const {return mType;};
```
返回当前输入类型
参数:无
返回:输入类型
---
### outputInfo
```cpp
Variable::Info* outputInfo(int index) const;
```
返回指定下标的输出信息
参数:
- `index` 下标值
返回:输出信息
---
### extra
```cpp
std::shared_ptr<BufferStorage> extra() const {
return mStorage;
};
```
返回附加信息
参数:无
返回:附加信息
---
### inside
```cpp
std::shared_ptr<Inside> inside() const {
return mInside;
};
```
返回内部信息
参数:无
返回:内部信息
---
### valid
```cpp
bool valid() const {
return mValid;
};
```
是否有效
参数:无
返回true/false