mxnet
Public Member Functions | Public Attributes | List of all members
mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst > Struct Template Reference

Broadcasting the tensor in the given axis. If keepdim is off, insert the broadcasting dim after axis. Otherwise broadcasting axis. More...

#include <broadcast_with_axis.h>

Inheritance diagram for mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >:
Inheritance graph
Collaboration diagram for mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >:
Collaboration graph

Public Member Functions

 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)
 

Public Attributes

const SrcExp & src_
 data oprand More...
 
index_t dst_last_
 size of the last dimension of dst More...
 
index_t trailing_
 product of the dimensions after the broadcasting axis More...
 
index_t size_
 new dimension of the broadcasting axis More...
 
index_t last_
 size of the last dimension of src More...
 
- Public Attributes inherited from mshadow::expr::MakeTensorExp< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, SrcExp, dimdst, DType >
Shape< dim > shape_
 the shape of this expression More...
 

Detailed Description

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
SrcExpsource expression
DTypedata type
dimsrcsource dimension
dimdstdestination dimension

Constructor & Destructor Documentation

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::BroadcastWithAxisExp ( const SrcExp &  src,
const int  axis,
const index_t  size 
)
inline

constructor

Member Data Documentation

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
index_t mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::dst_last_

size of the last dimension of dst

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
index_t mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::last_

size of the last dimension of src

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
index_t mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::size_

new dimension of the broadcasting axis

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
const SrcExp& mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::src_

data oprand

template<typename SrcExp, typename DType, int dimsrc, int dimdst>
index_t mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >::trailing_

product of the dimensions after the broadcasting axis


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