Broadcasting the tensor in the given axis. If keepdim is off, insert the broadcasting dim after axis. Otherwise broadcasting axis.
More...
|
| BroadcastWithAxisExp (const SrcExp &src, const int axis, const index_t size) |
|
Public Member Functions inherited from mshadow::expr::MakeTensorExp< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, SrcExp, dimdst, DType > |
const BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst > & | real_self (void) const |
| true self of subtype More...
|
|
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, SrcExp, dim, DType >, DType, type::kChainer > |
const MakeTensorExp< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, SrcExp, dim, DType > & | self (void) const |
|
MakeTensorExp< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, SrcExp, dim, DType > * | ptrself (void) |
|
template<typename SrcExp, typename DType, int dimsrc, int dimdst>
struct mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >
Broadcasting the tensor in the given axis. If keepdim is off, insert the broadcasting dim after axis. Otherwise broadcasting axis.
- Template Parameters
-
SrcExp | source expression |
DType | data type |
dimsrc | source dimension |
dimdst | destination dimension |