Create a one-hot indicator array. More...
#include <one_hot.h>
Inheritance diagram for mshadow::expr::OneHotEncodeExp< IndexExp, DType >:
Collaboration diagram for mshadow::expr::OneHotEncodeExp< IndexExp, DType >:
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... | |
Detailed Description
template<typename IndexExp, typename DType>
struct mshadow::expr::OneHotEncodeExp< IndexExp, DType >
Create a one-hot indicator array.
- Template Parameters
-
IndexExp type of index expression DType the type of elements
Constructor & Destructor Documentation
◆ OneHotEncodeExp()
template<typename IndexExp, typename DType>
|
inline |
constructor
Member Data Documentation
◆ index_
template<typename IndexExp, typename DType>
| const IndexExp& mshadow::expr::OneHotEncodeExp< IndexExp, DType >::index_ |
index operand
◆ num_choices_
template<typename IndexExp, typename DType>
| index_t mshadow::expr::OneHotEncodeExp< IndexExp, DType >::num_choices_ |
number of choices we can have.
The documentation for this struct was generated from the following file:
- /work/mxnet/3rdparty/mshadow/mshadow/extension/one_hot.h
