|
template<typename SrcExp , typename DType , int etype> |
BroadcastWithAxisExp< SrcExp, DType, ExpInfo< SrcExp >::kDim, ExpInfo< SrcExp >::kDim+1 > | mshadow::expr::broadcast_with_axis (const Exp< SrcExp, DType, etype > &src, const int axis, const index_t size) |
| Broadcasting the tensor after given axis. More...
|
|
template<typename SrcExp , typename DType , int etype> |
BroadcastWithAxisExp< SrcExp, DType, ExpInfo< SrcExp >::kDim, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_keepdim (const Exp< SrcExp, DType, etype > &src, const int axis, const index_t size) |
| Broadcasting the tensor in the given axis (keepdim turned on) More...
|
|
template<typename SrcExp , typename DType , int etype, typename TShape > |
BroadcastWithMultiAxesExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_multi_axes (const Exp< SrcExp, DType, etype > &src, const TShape &axes, const TShape &sizes) |
| Broadcasting the tensor in the given axis (keepdim turned on) More...
|
|
template<typename SrcExp , typename DType , int etype, typename TShape > |
BroadcastWithMultiAxesExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_to (const Exp< SrcExp, DType, etype > &src, const TShape &target_shape) |
| Broadcasting the tensor to the target shape, dimension of different sizes must be 1 in the original tensor. More...
|
|
Copyright (c) 2016 by Contributors
- Author
- Junyuan Xie, Xingjian Shi