|
mxnet
|
#include <tensor.h>


Public Member Functions | |
| DMLC_DECLARE_PARAMETER (SplitParam) | |
Public Attributes | |
| Tuple< int > | indices_or_sections |
| int | axis |
| bool | equal_split |
|
inline |
| int nnvm::top::SplitParam::axis |
| bool nnvm::top::SplitParam::equal_split |
| Tuple<int> nnvm::top::SplitParam::indices_or_sections |
1.8.11