|
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 |
1.8.11