mxnet
|
unpooling expr reverse operation of pooling, used to pass gradient back More...
#include <spatial_unpool.h>
Public Member Functions | |
UnPoolingExp (const SrcExp &data_src, const SrcExp &data_pooled, const SrcExp &grad_pooled, index_t ksize_y, index_t ksize_x, index_t kstride_y, index_t kstride_x) | |
constructor More... | |
Public Member Functions inherited from mshadow::expr::MakeTensorExp< UnPoolingExp< Reducer, SrcExp, DType, srcdim >, SrcExp, srcdim, DType > | |
const UnPoolingExp< Reducer, SrcExp, DType, srcdim > & | real_self (void) const |
true self of subtype More... | |
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< UnPoolingExp< Reducer, SrcExp, DType, srcdim >, SrcExp, dim, DType >, DType, type::kChainer > | |
const MakeTensorExp< UnPoolingExp< Reducer, SrcExp, DType, srcdim >, SrcExp, dim, DType > & | self (void) const |
MakeTensorExp< UnPoolingExp< Reducer, SrcExp, DType, srcdim >, SrcExp, dim, DType > * | ptrself (void) |
Public Attributes | |
const SrcExp & | data_src_ |
source input, corresponds to src in pooling More... | |
const SrcExp & | data_pooled_ |
result of pooled data, corresponds to result of pooling More... | |
const SrcExp & | grad_pooled_ |
gradient data of pooled part, to be propgate down More... | |
index_t | pshape_y_ |
shape of pooled expression More... | |
index_t | pshape_x_ |
shape of pooled expression More... | |
index_t | ksize_y_ |
kernel size in height More... | |
index_t | ksize_x_ |
kernel size in width More... | |
index_t | kstride_y_ |
kernel stride in y directory More... | |
index_t | kstride_x_ |
kernel stride in x directory More... | |
Public Attributes inherited from mshadow::expr::MakeTensorExp< UnPoolingExp< Reducer, SrcExp, DType, srcdim >, SrcExp, srcdim, DType > | |
Shape< dim > | shape_ |
the shape of this expression More... | |
unpooling expr reverse operation of pooling, used to pass gradient back
Reducer | reduction method during pooling |
SrcExp | source expression to be pooled from |
DType | the content data type |
srcdim | dimension of src |
|
inline |
constructor
const SrcExp& mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::data_pooled_ |
result of pooled data, corresponds to result of pooling
const SrcExp& mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::data_src_ |
source input, corresponds to src in pooling
const SrcExp& mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::grad_pooled_ |
gradient data of pooled part, to be propgate down
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::ksize_x_ |
kernel size in width
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::ksize_y_ |
kernel size in height
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::kstride_x_ |
kernel stride in x directory
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::kstride_y_ |
kernel stride in y directory
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::pshape_x_ |
shape of pooled expression
index_t mshadow::expr::UnPoolingExp< Reducer, SrcExp, DType, srcdim >::pshape_y_ |
shape of pooled expression