mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Classes | Namespaces | Macros | Typedefs | Enumerations | Functions
ndarray.h File Reference

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"
Include dependency graph for ndarray.h:
This graph shows which files directly or indirectly include this file:

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

Detailed Description

NDArray interface that handles array arithematics.

Copyright (c) 2015 by Contributors

Macro Definition Documentation

#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

REGISTER_NDARRAY_FUN(Plus)
.set_function(Plus);