Broadcasting the tensor in multiple axes. The dimension of the source tensor in the given axes must be 1. More...
#include <broadcast_with_axis.h>
Inheritance diagram for mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >:
Collaboration diagram for mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >:
Public Member Functions | |
| template<typename TShape > | |
| BroadcastWithMultiAxesExp (const SrcExp &src, const TShape &axes, const TShape &sizes) | |
 Public Member Functions inherited from mshadow::expr::MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dimsrc, DType > | |
| const BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc > & | real_self (void) const |
| true self of subtype More... | |
| Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType >, DType, type::kChainer > | |
| const MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > & | self (void) const |
| MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > * | ptrself (void) |
Public Attributes | |
| const SrcExp & | src_ |
| data oprand More... | |
| index_t | dst_last_ |
| size of the last dimension of dst More... | |
| index_t | axesnum_ |
| number of broadcasting axes More... | |
| Shape< dimsrc > | trailings_ |
| product of the dimensions after the broadcasting axses More... | |
| Shape< dimsrc > | sizes_ |
| new dimension of the broadcasting axes More... | |
| index_t | last_ |
| size of the last dimension of src More... | |
| Public Attributes inherited from mshadow::expr::MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dimsrc, DType > | |
| Shape< dim > | shape_ |
| the shape of this expression More... | |
Detailed Description
template<typename SrcExp, typename DType, int dimsrc>
struct mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >
Broadcasting the tensor in multiple axes. The dimension of the source tensor in the given axes must be 1.
- Template Parameters
-
SrcExp source expression DType data type dimsrc source dimension axesnum number of broadcasting dimensions
Constructor & Destructor Documentation
◆ BroadcastWithMultiAxesExp()
template<typename SrcExp, typename DType, int dimsrc>
template<typename TShape >
|
inline |
constructor
Member Data Documentation
◆ axesnum_
template<typename SrcExp, typename DType, int dimsrc>
| index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::axesnum_ |
number of broadcasting axes
◆ dst_last_
template<typename SrcExp, typename DType, int dimsrc>
| index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::dst_last_ |
size of the last dimension of dst
◆ last_
template<typename SrcExp, typename DType, int dimsrc>
| index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::last_ |
size of the last dimension of src
◆ sizes_
template<typename SrcExp, typename DType, int dimsrc>
| Shape<dimsrc> mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::sizes_ |
new dimension of the broadcasting axes
◆ src_
template<typename SrcExp, typename DType, int dimsrc>
| const SrcExp& mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::src_ |
data oprand
◆ trailings_
template<typename SrcExp, typename DType, int dimsrc>
| Shape<dimsrc> mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::trailings_ |
product of the dimensions after the broadcasting axses
The documentation for this struct was generated from the following file:
- /work/mxnet/3rdparty/mshadow/mshadow/extension/broadcast_with_axis.h
