|
mxnet
|
reduction to 1 dimension tensor input: Tensor<Device,k>: ishape output: Tensor<Device,1> shape[0] = ishape[dimkeep]; More...
#include <reduceto1d.h>


Public Member Functions | |
| ReduceTo1DExp (const SrcExp &src, DType scale) | |
| construct a repmat expression from src and nrow More... | |
Public Member Functions inherited from mshadow::expr::Exp< ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep >, DType, type::kComplex > | |
| const ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep > & | self (void) const |
| ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep > * | ptrself (void) |
Public Attributes | |
| const SrcExp & | src_ |
| source operand More... | |
| DType | scale_ |
| source operand, scale of the More... | |
reduction to 1 dimension tensor input: Tensor<Device,k>: ishape output: Tensor<Device,1> shape[0] = ishape[dimkeep];
| 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 |
|
inline |
construct a repmat expression from src and nrow
| DType mshadow::expr::ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep >::scale_ |
source operand, scale of the
| const SrcExp& mshadow::expr::ReduceTo1DExp< SrcExp, DType, Reducer, m_dimkeep >::src_ |
source operand
1.8.11