mxnet
Classes | Namespaces | Functions
broadcast_with_axis.h File Reference
#include <vector>
#include "../extension.h"
Include dependency graph for broadcast_with_axis.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

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. More...
 
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. More...
 
struct  mshadow::expr::Plan< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, DType >
 
struct  mshadow::expr::Plan< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, DType >
 

Namespaces

 mshadow
 namespace for mshadow
 
 mshadow::expr
 namespace for abstract expressions and expressions template, have no dependency on tensor.h, These data structure takes no charge in computations, they are only used to define operations and represent expression in a symbolic way
 

Functions

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...
 

Detailed Description

Copyright (c) 2016 by Contributors

Author
Junyuan Xie, Xingjian Shi