mxnet
|
Create a one-hot indicator array. More...
#include <one_hot.h>
Public Member Functions | |
OneHotEncodeExp (const IndexExp &index, index_t num_choices) | |
constructor More... | |
Public Member Functions inherited from mshadow::expr::Exp< OneHotEncodeExp< IndexExp, DType >, DType, type::kChainer > | |
const OneHotEncodeExp< IndexExp, DType > & | self (void) const |
OneHotEncodeExp< IndexExp, DType > * | ptrself (void) |
Public Attributes | |
const IndexExp & | index_ |
index operand More... | |
index_t | num_choices_ |
number of choices we can have. More... | |
Create a one-hot indicator array.
IndexExp | type of index expression |
DType | the type of elements |
|
inline |
constructor
const IndexExp& mshadow::expr::OneHotEncodeExp< IndexExp, DType >::index_ |
index operand
index_t mshadow::expr::OneHotEncodeExp< IndexExp, DType >::num_choices_ |
number of choices we can have.