ocean::Filler
大约 5 分钟
ocean::Filler
张量填充器 More...
#include <Filler.hpp>
Public Functions
Name | |
---|---|
Filler() 空填充器 | |
Filler(PIFiller filler_) 构造填充器 | |
Filler(const Filler & arg) 拷贝构造 | |
Filler & | operator=(const Filler & arg) 赋值构造 |
Filler(const FillerType & type, const DataType & dataType, const FillerArguments & args) 从数值参数构造填充器 | |
Filler(const FillerType & type, const DataType & dataType, const Arguments & args) 从通用参数构造填充器 | |
Filler(const DataType & dataType) 构建零填充器 | |
Filler(const Tensor & copy) 构建拷贝填充器 | |
Filler(VariableValue & value) 从变量值构建张量填充器 | |
template <typename T > | Filler(T value) 从数值构建张量填充器 |
template <typename T > | Filler(T min, T max, uint32_t seed =0) 构建均匀分布填充器 |
Filler(DataType type, float mean, float stddev, uint32_t seed =0) 构建正态分布填充器 | |
virtual | ~Filler() 析构填充器 |
const PIFiller & | getFiller() const 获取内部填充器指针 |
void | setFiller(PIFiller filler) 设置填充器指针 |
Filler | makeZero(const DataType & dataType) 构建零填充器 |
Filler | makeCopy(const Tensor & copy) 构建拷贝填充器 |
Filler | makeConst(VariableValue & value) 从变量值构建张量填充器 |
template <typename T > Filler | makeConst(T value) 从数值构建张量填充器 |
template <typename T > Filler | makeUniform(T min, T max, uint32_t seed =0) 构建均匀分布填充器 |
Filler | makeLogNormal(DataType type, float mean, float stddev, uint32_t seed =0) 构建正态分布填充器 |
Filler | makeLocalConvolutionFiller(uint32_t receptorField, uint32_t width, uint32_t height, Tensor scaled) 构建局部卷积填充器 |
Protected Functions
Name | |
---|---|
void | makeFiller(const FillerType & type, const DataType & dataType, const Arguments & args) 从通用参数构造填充器 |
void | makeFiller(const FillerType & type, const DataType & dataType, const FillerArguments & args) 从数值参数构造填充器 |
Protected Attributes
Name | |
---|---|
PIFiller | filler 填充器 |
Detailed Description
class ocean::Filler;
张量填充器
张量填充器统一了张量变量的初始化机制,提供了一种异步填充机制,可以用于初始化或重置张量的内容。
- 常量填充, 使用一个固定值填充张量
- 拷贝填充, 使用等效的Tensor进行填充
- 均匀随机填充, 指定数据范围随机均匀填充
- 正态随机填充, 指定均值及方差进行正态随机填充
- 局部卷积填充, 通过指定连接位置范围进行对应位置的均匀填充
- 填充器可以按需扩展,灵活方便
Public Functions Documentation
function Filler
Filler()
空填充器
function Filler
explicit Filler(
PIFiller filler_
)
构造填充器
Parameters:
- filler_ 内部填充器指针
Note: 该方法仅被内部模块使用
function Filler
Filler(
const Filler & arg
)
拷贝构造
Parameters:
- arg
function operator=
Filler & operator=(
const Filler & arg
)
赋值构造
Parameters:
- arg
Return:
function Filler
Filler(
const FillerType & type,
const DataType & dataType,
const FillerArguments & args
)
从数值参数构造填充器
Parameters:
- type 填充器类型
- dataType 数据类型
- args 数值参数列表
function Filler
Filler(
const FillerType & type,
const DataType & dataType,
const Arguments & args
)
从通用参数构造填充器
Parameters:
- type 填充器类型
- dataType 数据类型
- args 通用参数列表
function Filler
Filler(
const DataType & dataType
)
构建零填充器
Parameters:
- dataType 数据类型
function Filler
Filler(
const Tensor & copy
)
构建拷贝填充器
Parameters:
- copy 待拷贝张量
function Filler
Filler(
VariableValue & value
)
从变量值构建张量填充器
Parameters:
- value 变量值
function Filler
template <typename T >
Filler(
T value
)
从数值构建张量填充器
Parameters:
- value 数值
Template Parameters:
- T 数据类型
function Filler
template <typename T >
Filler(
T min,
T max,
uint32_t seed =0
)
构建均匀分布填充器
Parameters:
- min 最大值
- max 最小值
- seed 随机种子
Template Parameters:
- T 数据类型
function Filler
Filler(
DataType type,
float mean,
float stddev,
uint32_t seed =0
)
构建正态分布填充器
Parameters:
- type 数据类型
- mean 均值
- stddev 标准差
- seed 随机种子
function ~Filler
virtual ~Filler()
析构填充器
function getFiller
const PIFiller & getFiller() const
获取内部填充器指针
Return: 内部填充器指针
function setFiller
void setFiller(
PIFiller filler
)
设置填充器指针
Parameters:
- filler
function makeZero
static Filler makeZero(
const DataType & dataType
)
构建零填充器
Parameters:
- dataType 数据类型
Return: 填充器
function makeCopy
static Filler makeCopy(
const Tensor & copy
)
构建拷贝填充器
Parameters:
- copy 待拷贝张量
Return: 填充器
function makeConst
static Filler makeConst(
VariableValue & value
)
从变量值构建张量填充器
Parameters:
- value 变量值
Return: 填充器
function makeConst
template <typename T >
static Filler makeConst(
T value
)
从数值构建张量填充器
Parameters:
- value 数值
Template Parameters:
- T 数据类型
Return: 填充器
function makeUniform
template <typename T >
static Filler makeUniform(
T min,
T max,
uint32_t seed =0
)
构建均匀分布填充器
Parameters:
- min 最大值
- max 最小值
- seed 随机种子
Template Parameters:
- T 数据类型
Return: 填充器
function makeLogNormal
static Filler makeLogNormal(
DataType type,
float mean,
float stddev,
uint32_t seed =0
)
构建正态分布填充器
Parameters:
- type 数据类型
- mean 均值
- stddev 标准差
- seed 随机种子
Return: 填充器
function makeLocalConvolutionFiller
static Filler makeLocalConvolutionFiller(
uint32_t receptorField,
uint32_t width,
uint32_t height,
Tensor scaled
)
构建局部卷积填充器
Parameters:
- receptorField 感受野
- width 输入宽度
- height 输入高度
- scaled
Return: 填充器
Protected Functions Documentation
function makeFiller
void makeFiller(
const FillerType & type,
const DataType & dataType,
const Arguments & args
)
从通用参数构造填充器
Parameters:
- type 填充器类型
- dataType 数据类型
- args 通用参数列表
function makeFiller
void makeFiller(
const FillerType & type,
const DataType & dataType,
const FillerArguments & args
)
从数值参数构造填充器
Parameters:
- type 填充器类型
- dataType 数据类型
- args 数值参数列表
Protected Attributes Documentation
variable filler
PIFiller filler;
填充器
Updated on 2022-08-19 at 17:14:26 +0800