mxnet
|
NDArray interface that handles array arithematics. More...
#include <dmlc/base.h>
#include <dmlc/logging.h>
#include <dmlc/io.h>
#include <dmlc/type_traits.h>
#include <dmlc/registry.h>
#include <vector>
#include <map>
#include <string>
#include <memory>
#include "./base.h"
#include "./storage.h"
#include "./engine.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 | |
mxnet | |
namespace of mxnet | |
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::NDArrayFunctionTypeMask { mxnet::kNDArrayArgBeforeScalar = 1, mxnet::kScalarArgBeforeNDArray = 1 << 1, mxnet::kAcceptEmptyMutateTarget = 1 << 2 } |
mask information on how functions can be exposed More... | |
Functions | |
void | mxnet::CopyFromTo (const NDArray &from, 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::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 substraction More... | |
NDArray | mxnet::operator- (const NDArray &lhs, const real_t &rhs) |
elementwise substraction 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 the random number generator. 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... | |
dmlc::DMLC_DECLARE_TRAITS (has_saveload, mxnet::NDArray, true) | |
traits More... | |
NDArray interface that handles array arithematics.
Copyright (c) 2015 by Contributors
#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