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