reduction to 1 dimension tensor input: Tensor<Device,k>: ishape output: Tensor<Device,1> shape[0] = ishape[dimkeep];
More...
template<typename SrcExp, typename DType, typename Reducer, int m_dimkeep>
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];
- Template Parameters
-
SrcExp | type of expression to be reduced |
DType | the data type of the scalar |
Reducer | which reducer to use |
m_dimkeep | which dimension to be kept, encoded with dimsrc - dimkeep |