mxnet
|
Calculate embedding gradient. More...
#include <take_grad.h>
Public Member Functions | |
TakeGradExp (const IndexExp &index, const SrcExp &src, const index_t input_dim) | |
constructor More... | |
Public Member Functions inherited from mshadow::expr::Exp< TakeGradExp< IndexExp, SrcExp, DType >, DType, type::kChainer > | |
const TakeGradExp< IndexExp, SrcExp, DType > & | self (void) const |
TakeGradExp< IndexExp, SrcExp, DType > * | ptrself (void) |
Public Attributes | |
const IndexExp & | index_ |
index oprand More... | |
const SrcExp & | src_ |
out gradient oprand More... | |
const index_t | input_dim_ |
batch size More... | |
Calculate embedding gradient.
IndexExp | type of index expression |
SrcExp | type of src expression |
DType | data type |
|
inline |
constructor
const IndexExp& mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::index_ |
index oprand
const index_t mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::input_dim_ |
batch size
const SrcExp& mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::src_ |
out gradient oprand