mxnet
|
support for sum_rows and sumall_except_dim More...
#include "../extension.h"
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... | |
support for sum_rows and sumall_except_dim
Copyright (c) 2014 by Contributors