7 #ifndef MSHADOW_EXTENSION_RANGE_H_ 8 #define MSHADOW_EXTENSION_RANGE_H_ 10 #include "../extension.h" 23 template<
typename DType>
25 public Exp<RangeExp<DType>, DType, type::kMapper> {
31 RangeExp(DType start, DType stop, DType step,
int repeat)
32 : start_(start), stop_(stop), step_(step), repeat_(repeat) {}
35 template<
typename DType>
37 range(DType start, DType stop, DType step = 1,
int repeat = 1) {
44 template<
typename DType>
64 template<
typename DType>
71 template<
typename DType>
72 inline int RangeOutSize(DType start, DType stop, DType step,
int repeat) {
73 return repeat * ((stop - start - 1) / step + 1);
78 double d_start =
static_cast<double>(start);
79 double d_stop =
static_cast<double>(stop);
80 double d_step =
static_cast<double>(step);
81 return repeat *
static_cast<int>(ceil((d_stop - d_start) / d_step));
86 return repeat *
static_cast<int>(ceil((stop - start) / step));
90 template<
int dim,
typename DType>
95 <<
"RangeExp only support 1 dimension output, received " << dim;
97 <<
"RangeExp does not support step=0, received " << t.
step_;
99 <<
"RangeExp only supports repeat > 0, received " << t.
repeat_;
101 CHECK(t.
start_ < t.
stop_) <<
"RangeExp does not support (start, stop, step) = " 104 CHECK(t.
start_ > t.
stop_) <<
"RangeExp does not support (start, stop, step)= " 111 template<
typename DType>
113 static const int kDim = 1;
114 static const int kDevMask = 0xffff;
118 #endif // MSHADOW_EXTENSION_RANGE_H_ Definition: expr_engine-inl.h:40
int RangeOutSize(DType start, DType stop, DType step, int repeat)
Definition: range.h:72
MSHADOW_XINLINE DType Eval(index_t y, index_t x) const
Definition: range.h:53
Plan(const RangeExp< DType > &e)
Definition: range.h:47
const DType start_
Definition: range.h:26
#define MSHADOW_XINLINE
Definition: base.h:204
static type inference template, used to get the dimension of each expression, if ExpInfo<E>::kDim == ...
Definition: expr_engine-inl.h:244
const int repeat_
Definition: range.h:29
const DType stop_
Definition: range.h:27
int32_t index_t
type that will be used for index
Definition: base.h:291
MSHADOW_XINLINE Shape< 1 > Shape1(index_t s0)
construct a one dimension shape, stride will equal s0
Definition: tensor.h:188
runtime shape checking template get the shape of an expression, report error if shape mismatch ...
Definition: expr_engine-inl.h:346
const DType step_
Definition: range.h:28
RangeExp< DType > range(DType start, DType stop, DType step=1, int repeat=1)
Definition: range.h:37
int RangeOutSize< double >(double start, double stop, double step, int repeat)
Definition: range.h:85
defines how expression exp can be evaluated and stored into dst
Definition: expression.h:61
Plan< BinaryMapExp< OP, TA, TB, DType, etype >, DType > MakePlan(const BinaryMapExp< OP, TA, TB, DType, etype > &e)
Definition: expr_engine-inl.h:221
RangeExp(DType start, DType stop, DType step, int repeat)
constructor
Definition: range.h:31
Generate a range vector similar to python: range(start, stop[, step][, repeat]). If step is positive...
Definition: range.h:24
namespace for mshadow
Definition: base.h:282
static Shape< dim > Check(const RangeExp< DType > &t)
Definition: range.h:93
int RangeOutSize< float >(float start, float stop, float step, int repeat)
Definition: range.h:77