14 namespace dataloader {
25 num_batches = unsigned(x->
get_shape(0) / batch_size);
26 assert(num_batches > 0);
27 assert(num_batches ==
unsigned(y->
get_shape(0) / batch_size));
41 virtual void reset() = 0;
43 virtual unsigned int get_batch_size()
const {
return batch_size;}
44 virtual void set_batch_size(
unsigned int size) {
46 num_batches = unsigned(x->get_shape(0) / batch_size);
47 assert(num_batches > 0);
48 assert(num_batches ==
unsigned(y->get_shape(0) / batch_size));
50 virtual unsigned int get_num_batches()
const {
return num_batches;}
55 unsigned int batch_size;
56 unsigned int sample_size_x;
57 unsigned int sample_size_y;
58 unsigned int num_batches;
std::vector< unsigned int > get_shape() const
Definition: tensor.h:181
unsigned int get_size() const
Definition: tensor.h:192
Definition: dataloader.h:17
virtual void next(Tensor< T > *x_batch, Tensor< T > *y_batch)=0
DataLoader(Tensor< T > *x, Tensor< T > *y, unsigned int batch_size)
Definition: dataloader.h:24