mxnet
|
#include <tensor.h>
Public Member Functions | |
DMLC_DECLARE_PARAMETER (ReduceParam) | |
Public Attributes | |
TShape | axis |
bool | keepdims |
bool | exclude |
int | dtype |
|
inline |
TShape nnvm::top::ReduceParam::axis |
int nnvm::top::ReduceParam::dtype |
bool nnvm::top::ReduceParam::exclude |
bool nnvm::top::ReduceParam::keepdims |