26 #ifndef MSHADOW_EXTENSION_RANGE_H_ 27 #define MSHADOW_EXTENSION_RANGE_H_ 29 #include "../extension.h" 42 template<
typename DType>
44 public Exp<RangeExp<DType>, DType, type::kMapper> {
50 RangeExp(DType start, DType stop, DType step,
int repeat)
51 : start_(start), stop_(stop), step_(step), repeat_(repeat) {}
54 template<
typename DType>
56 range(DType start, DType stop, DType step = 1,
int repeat = 1) {
63 template<
typename DType>
83 template<
typename DType>
90 template<
typename DType>
91 inline int RangeOutSize(DType start, DType stop, DType step,
int repeat) {
92 return repeat * ((stop - start - 1) / step + 1);
97 double d_start =
static_cast<double>(start);
98 double d_stop =
static_cast<double>(stop);
99 double d_step =
static_cast<double>(step);
100 return repeat *
static_cast<int>(ceil((d_stop - d_start) / d_step));
105 return repeat *
static_cast<int>(ceil((stop - start) / step));
109 template<
int dim,
typename DType>
114 <<
"RangeExp only support 1 dimension output, received " << dim;
116 <<
"RangeExp does not support step=0, received " << t.
step_;
118 <<
"RangeExp only supports repeat > 0, received " << t.
repeat_;
120 CHECK(t.
start_ < t.
stop_) <<
"RangeExp does not support (start, stop, step) = " 123 CHECK(t.
start_ > t.
stop_) <<
"RangeExp does not support (start, stop, step)= " 130 template<
typename DType>
132 static const int kDim = 1;
133 static const int kDevMask = 0xffff;
137 #endif // MSHADOW_EXTENSION_RANGE_H_ Definition: expr_engine-inl.h:59
int RangeOutSize(DType start, DType stop, DType step, int repeat)
Definition: range.h:91
MSHADOW_XINLINE DType Eval(index_t y, index_t x) const
Definition: range.h:72
Plan(const RangeExp< DType > &e)
Definition: range.h:66
const DType start_
Definition: range.h:45
#define MSHADOW_XINLINE
Definition: base.h:223
static type inference template, used to get the dimension of each expression, if ExpInfo<E>::kDim == ...
Definition: expr_engine-inl.h:263
const int repeat_
Definition: range.h:48
const DType stop_
Definition: range.h:46
int32_t index_t
type that will be used for index
Definition: base.h:336
MSHADOW_XINLINE Shape< 1 > Shape1(index_t s0)
construct a one dimension shape, stride will equal s0
Definition: tensor.h:207
runtime shape checking template get the shape of an expression, report error if shape mismatch ...
Definition: expr_engine-inl.h:365
const DType step_
Definition: range.h:47
RangeExp< DType > range(DType start, DType stop, DType step=1, int repeat=1)
Definition: range.h:56
int RangeOutSize< double >(double start, double stop, double step, int repeat)
Definition: range.h:104
defines how expression exp can be evaluated and stored into dst
Definition: expression.h:80
Plan< BinaryMapExp< OP, TA, TB, DType, etype >, DType > MakePlan(const BinaryMapExp< OP, TA, TB, DType, etype > &e)
Definition: expr_engine-inl.h:240
RangeExp(DType start, DType stop, DType step, int repeat)
constructor
Definition: range.h:50
Generate a range vector similar to python: range(start, stop[, step][, repeat]). If step is positive...
Definition: range.h:43
overloaded + operator between half_t and bf16_t
Definition: base.h:327
static Shape< dim > Check(const RangeExp< DType > &t)
Definition: range.h:112
int RangeOutSize< float >(float start, float stop, float step, int repeat)
Definition: range.h:96