mxnet
|
#include <operator.h>
Public Member Functions | |
Operator (const std::string &operator_name) | |
Operator constructor. More... | |
Operator & | operator= (const Operator &rhs) |
template<typename T > | |
Operator & | SetParam (const std::string &name, const T &value) |
set config parameters More... | |
template<typename T > | |
Operator & | SetParam (int pos, const T &value) |
set config parameters from positional inputs More... | |
Operator & | SetInput (const std::string &name, const Symbol &symbol) |
add an input symbol More... | |
template<int N = 0> | |
void | PushInput (const Symbol &symbol) |
add an input symbol More... | |
Operator & | operator() () |
add input symbols More... | |
Operator & | operator() (const Symbol &symbol) |
add input symbols More... | |
Operator & | operator() (const std::vector< Symbol > &symbols) |
add a list of input symbols More... | |
Symbol | CreateSymbol (const std::string &name="") |
create a Symbol from the current operator More... | |
Operator & | SetInput (const std::string &name, const NDArray &ndarray) |
add an input ndarray More... | |
template<int N = 0> | |
Operator & | PushInput (const NDArray &ndarray) |
add an input ndarray More... | |
template<class T , class... Args, int N = 0> | |
Operator & | PushInput (const T &t, Args...args) |
add positional inputs More... | |
template<class T , int N = 0> | |
Operator & | PushInput (const T &t) |
add the last positional input More... | |
Operator & | operator() (const NDArray &ndarray) |
add input ndarrays More... | |
Operator & | operator() (const std::vector< NDArray > &ndarrays) |
add a list of input ndarrays More... | |
template<typename... Args> | |
Operator & | operator() (Args...args) |
add input ndarrays More... | |
std::vector< NDArray > | Invoke () |
void | Invoke (NDArray &output) |
void | Invoke (std::vector< NDArray > &outputs) |
Operator interface.
|
explicit |
Operator constructor.
operator_name | type of the operator |
Symbol mxnet::cpp::Operator::CreateSymbol | ( | const std::string & | name = "" | ) |
std::vector<NDArray> mxnet::cpp::Operator::Invoke | ( | ) |
void mxnet::cpp::Operator::Invoke | ( | NDArray & | output | ) |
void mxnet::cpp::Operator::Invoke | ( | std::vector< NDArray > & | outputs | ) |
|
inline |
add input symbols
add input symbols
symbol | the input symbol |
add a list of input symbols
symbols | the vector of the input symbols |
add input ndarrays
ndarray | the input ndarray |
add a list of input ndarrays
ndarrays | the vector of the input ndarrays |
|
inline |
add input ndarrays
|
inline |
add an input symbol
symbol | the input symbol |
add an input ndarray
ndarray | the input ndarray |
|
inline |
add positional inputs
|
inline |
add the last positional input
add an input symbol
name | name of the input symbol |
symbol | the input symbol |
add an input ndarray
name | name of the input ndarray |
ndarray | the input ndarray |
|
inline |
set config parameters
name | name of the config parameter |
value | value of the config parameter |
|
inline |
set config parameters from positional inputs
pos | the position of parameter |
value | value of the config parameter |