mxnet
extension.h
Go to the documentation of this file.
1 
8 #ifndef MSHADOW_EXTENSION_H_
9 #define MSHADOW_EXTENSION_H_
10 #include "./expr_engine-inl.h"
11 #include "./extension/broadcast.h"
14 #include "./extension/reshape.h"
15 #include "./extension/swapaxis.h"
16 #include "./extension/reduceto1d.h"
21 #include "./extension/pad.h"
22 #include "./extension/crop.h"
23 #include "./extension/mirror.h"
24 #include "./extension/concat.h"
26 #include "./extension/choose.h"
27 #include "./extension/fill.h"
28 #include "./extension/one_hot.h"
29 #include "./extension/slice.h"
30 #include "./extension/slice_ex.h"
31 #include "./extension/take.h"
32 #include "./extension/take_grad.h"
36 #include "./extension/transpose.h"
37 #include "./extension/flip.h"
38 #include "./extension/complex.h"
39 #include "./extension/range.h"
40 #include "./extension/mask.h"
41 #endif // MSHADOW_EXTENSION_H_
support for chpool
support for implicit array filling operation
support for complex operations
support for reshape
support for transpose
support for unpack
support for mirror
support for implicit GEMM operation
support for pad
support for crop
support for implicit array selection operation
support for unpool
support generating a range vector
definitions of how expressions should be evaluated
support for concatenation
support for sum_rows and sumall_except_dim
support for pack
Create one-hot indicator array based on the index.
support for spatial pooling
support for swapaxis
support for slice a certain dimension.
support for broadcast and repmat
support for flip a certain dimension.