| 533 | |
| 534 | template <typename Dtype> |
| 535 | int DataTransformer<Dtype>::Rand(int n) { |
| 536 | CHECK(rng_); |
| 537 | CHECK_GT(n, 0); |
| 538 | caffe::rng_t* rng = |
| 539 | static_cast<caffe::rng_t*>(rng_->generator()); |
| 540 | return ((*rng)() % n); |
| 541 | } |
| 542 | |
| 543 | INSTANTIATE_CLASS(DataTransformer); |
| 544 |