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

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 >:
Inheritance graph
Collaboration diagram for mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >:
Collaboration graph

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
SrcExpsource expression
DTypedata type
dimsrcsource dimension
axesnumnumber of broadcasting dimensions

Constructor & Destructor Documentation

template<typename SrcExp, typename DType, int dimsrc>
template<typename TShape >
mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::BroadcastWithMultiAxesExp ( const SrcExp &  src,
const TShape &  axes,
const TShape &  sizes 
)
inline

constructor

Member Data Documentation

template<typename SrcExp, typename DType, int dimsrc>
index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::axesnum_

number of broadcasting axes

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

size of the last dimension of dst

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

size of the last dimension of src

template<typename SrcExp, typename DType, int dimsrc>
Shape<dimsrc> mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::sizes_

new dimension of the broadcasting axes

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

data oprand

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: