mxnet
|
reduce out the dimension of src labeled by axis. More...
#include <reduce_with_axis.h>
Public Member Functions | |
ReduceWithAxisExp (const SrcExp &src, int axis) | |
Public Member Functions inherited from mshadow::expr::MakeTensorExp< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, SrcExp, dimdst, DType > | |
const ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst > & | real_self (void) const |
true self of subtype More... | |
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, SrcExp, dim, DType >, DType, type::kChainer > | |
const MakeTensorExp< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, SrcExp, dim, DType > & | self (void) const |
MakeTensorExp< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, SrcExp, dim, DType > * | ptrself (void) |
Public Attributes | |
const SrcExp & | src_ |
source oprand More... | |
index_t | last_dst_dim_ |
size of last destination dimension More... | |
index_t | trailing_ |
size of trailing dimensions More... | |
index_t | size_ |
size of axis dimension More... | |
index_t | last_ |
size of last src dimension More... | |
Public Attributes inherited from mshadow::expr::MakeTensorExp< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, SrcExp, dimdst, DType > | |
Shape< dim > | shape_ |
the shape of this expression More... | |
reduce out the dimension of src labeled by axis.
Reducer | type of reducer |
SrcExp | type of source expression |
DType | data type |
|
inlineexplicit |
constructor
index_t mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >::last_ |
size of last src dimension
index_t mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >::last_dst_dim_ |
size of last destination dimension
index_t mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >::size_ |
size of axis dimension
const SrcExp& mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >::src_ |
source oprand
index_t mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >::trailing_ |
size of trailing dimensions