mxnet
Classes | Namespaces | Functions
reduceto1d.h File Reference

support for sum_rows and sumall_except_dim More...

#include "../extension.h"
Include dependency graph for reduceto1d.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mshadow::expr::ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep >
 reduction to 1 dimension tensor input: Tensor<Device,k>: ishape output: Tensor<Device,1> shape[0] = ishape[dimkeep]; More...
 
struct  mshadow::expr::ExpComplexEngine< SV, Tensor< Device, 1, DType >, ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep >, DType >
 
struct  mshadow::expr::ExpComplexEngine< SV, Tensor< Device, 1, DType >, ReduceTo1DExp< SrcExp, DType, Reducer, 1 >, 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<int dimkeep, typename SrcExp , typename DType , int etype>
ReduceTo1DExp< SrcExp, DType, red::sum, ExpInfo< SrcExp >::kDim-dimkeep > mshadow::expr::sumall_except_dim (const Exp< SrcExp, DType, etype > &exp)
 a sum over all dimensions, except dimkeep More...
 
template<int dimkeep, typename Reducer , typename SrcExp , typename DType , int etype>
ReduceTo1DExp< SrcExp, DType, Reducer, ExpInfo< SrcExp >::kDim-dimkeep > mshadow::expr::reduce_except_dim (const Exp< SrcExp, DType, etype > &exp)
 reduce over all dimensions, except dimkeep More...
 
template<typename SrcExp , typename DType , int etype>
ReduceTo1DExp< SrcExp, DType, red::sum, 1 > mshadow::expr::sum_rows (const Exp< SrcExp, DType, etype > &exp)
 a expression that sum over rows of a matrix More...
 

Detailed Description

support for sum_rows and sumall_except_dim

Copyright (c) 2014 by Contributors

Author
Tianqi Chen