mxnet
|
NDArray interface that handles array arithematics. More...
#include <dmlc/base.h>
#include <dmlc/io.h>
#include <dmlc/logging.h>
#include <dmlc/registry.h>
#include <dmlc/type_traits.h>
#include <nnvm/node.h>
#include <algorithm>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include "./base.h"
#include "./engine.h"
#include "./storage.h"
Go to the source code of this file.
Classes | |
class | mxnet::NDArray |
ndarray interface More... | |
struct | mxnet::NDArrayFunctionReg |
Registry entry for NDArrayFunction. More... | |
Namespaces | |
dnnl | |
mxnet | |
namespace of mxnet | |
mxnet::csr | |
mxnet::rowsparse | |
dmlc | |
namespace for dmlc | |
Macros | |
#define | MXNET_REGISTER_NDARRAY_FUN(name) DMLC_REGISTRY_REGISTER(::mxnet::NDArrayFunctionReg, NDArrayFunctionReg, name) |
Macro to register NDArray function. More... | |
Typedefs | |
typedef std::function< void(NDArray **used_vars, real_t *scalars, NDArray **mutate_vars, int num_params, char **param_keys, char **param_vals)> | mxnet::NDArrayAPIFunction |
definition of NDArray function More... | |
Enumerations | |
enum | mxnet::csr::CSRAuxType { mxnet::csr::kIndPtr, mxnet::csr::kIdx } |
enum | mxnet::rowsparse::RowSparseAuxType { mxnet::rowsparse::kIdx } |
enum | mxnet::NDArrayStorageType { mxnet::kUndefinedStorage = -1, mxnet::kDefaultStorage, mxnet::kRowSparseStorage, mxnet::kCSRStorage } |
enum | mxnet::NDArrayFormatErr { mxnet::kNormalErr, mxnet::kCSRShapeErr, mxnet::kCSRIndPtrErr, mxnet::kCSRIdxErr, mxnet::kRSPShapeErr, mxnet::kRSPIdxErr } |
enum | mxnet::NDArrayFunctionTypeMask { mxnet::kNDArrayArgBeforeScalar = 1, mxnet::kScalarArgBeforeNDArray = 1 << 1, mxnet::kAcceptEmptyMutateTarget = 1 << 2 } |
mask information on how functions can be exposed More... | |
Functions | |
size_t | mxnet::num_aux_data (NDArrayStorageType stype) |
void | mxnet::CopyFromTo (const NDArray &from, const NDArray *to, int priority=0) |
issue an copy operation from one NDArray to another the two ndarray can sit on different devices this operation will be scheduled by the engine More... | |
void | mxnet::CopyFromTo (const NDArray &from, const NDArray &to, int priority=0, bool is_opr=false) |
issue an copy operation from one NDArray to another the two ndarray can sit on different devices this operation will be scheduled by the engine More... | |
void | mxnet::ElementwiseSum (const std::vector< NDArray > &source, NDArray *out, int priority=0) |
Perform elementwise sum over each data from source, store result into out. More... | |
NDArray | mxnet::operator+ (const NDArray &lhs, const NDArray &rhs) |
elementwise add More... | |
NDArray | mxnet::operator+ (const NDArray &lhs, const real_t &rhs) |
elementwise add More... | |
NDArray | mxnet::operator- (const NDArray &lhs, const NDArray &rhs) |
elementwise subtraction More... | |
NDArray | mxnet::operator- (const NDArray &lhs, const real_t &rhs) |
elementwise subtraction More... | |
NDArray | mxnet::operator* (const NDArray &lhs, const NDArray &rhs) |
elementwise multiplication More... | |
NDArray | mxnet::operator* (const NDArray &lhs, const real_t &rhs) |
elementwise multiplication More... | |
NDArray | mxnet::operator/ (const NDArray &lhs, const NDArray &rhs) |
elementwise division More... | |
NDArray | mxnet::operator/ (const NDArray &lhs, const real_t &rhs) |
elementwise division More... | |
void | mxnet::RandomSeed (uint32_t seed) |
Seed all random number generator in mxnet. More... | |
void | mxnet::RandomSeed (Context ctx, uint32_t seed) |
Seed the random number generator of the device. More... | |
void | mxnet::SampleUniform (real_t begin, real_t end, NDArray *out) |
Sample uniform distribution for each elements of out. More... | |
void | mxnet::SampleGaussian (real_t mu, real_t sigma, NDArray *out) |
Sample gaussian distribution for each elements of out. More... | |
void | mxnet::SampleGamma (real_t alpha, real_t beta, NDArray *out) |
Sample gamma distribution for each elements of out. More... | |
void | mxnet::SampleExponential (real_t lambda, NDArray *out) |
Sample exponential distribution for each elements of out. More... | |
void | mxnet::SamplePoisson (real_t lambda, NDArray *out) |
Sample Poisson distribution for each elements of out. More... | |
void | mxnet::SampleNegBinomial (int32_t k, real_t p, NDArray *out) |
Sample negative binomial distribution for each elements of out. More... | |
void | mxnet::SampleGenNegBinomial (real_t mu, real_t alpha, NDArray *out) |
Sample generalized negative binomial distribution for each elements of out. More... | |
dmlc::DMLC_DECLARE_TRAITS (has_saveload, mxnet::NDArray, true) | |
traits More... | |
NDArray interface that handles array arithematics.
#define MXNET_REGISTER_NDARRAY_FUN | ( | name | ) | DMLC_REGISTRY_REGISTER(::mxnet::NDArrayFunctionReg, NDArrayFunctionReg, name) |
Macro to register NDArray function.
Example: the following code is example to register a plus