mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType > Struct Template Reference

Calculate embedding gradient. More...

#include <take_grad.h>

Inheritance diagram for mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >:

Collaboration diagram for mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >:

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...

Detailed Description

template<typename IndexExp, typename SrcExp, typename DType>
struct mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >

Calculate embedding gradient.

Template Parameters

IndexExp	type of index expression
SrcExp	type of src expression
DType	data type

Constructor & Destructor Documentation

template<typename IndexExp, typename SrcExp, typename DType>

mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::TakeGradExp ( const IndexExp &  index, const SrcExp &  src, const index_t  input_dim  )

inline

constructor

Member Data Documentation

template<typename IndexExp, typename SrcExp, typename DType>

const IndexExp& mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::index_

index oprand

template<typename IndexExp, typename SrcExp, typename DType>

const index_t mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::input_dim_

batch size

template<typename IndexExp, typename SrcExp, typename DType>

const SrcExp& mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >::src_

out gradient oprand

---

The documentation for this struct was generated from the following file:

• /work/mxnet/3rdparty/mshadow/mshadow/extension/take_grad.h