| 548 | |
| 549 | template <typename Dtype> |
| 550 | int DataTransformer<Dtype>::Rand(int n) { |
| 551 | CHECK(rng_); |
| 552 | CHECK_GT(n, 0); |
| 553 | caffe::rng_t* rng = |
| 554 | static_cast<caffe::rng_t*>(rng_->generator()); |
| 555 | return ((*rng)() % n); |
| 556 | } |
| 557 | |
| 558 | INSTANTIATE_CLASS(DataTransformer); |
| 559 |