linearloader.h

Go to the documentation of this file.

#pragma once
#include "dataloader/dataloader.h"

namespace magmadnn {
namespace dataloader {

template <typename T>
class LinearLoader : public DataLoader<T> {
public:
    LinearLoader(Tensor<T> *x, Tensor<T> *y, unsigned int batch_size);
    
    virtual void next(Tensor<T> *x_batch, Tensor<T> *y_batch);

    virtual void reset();
    
private:
    unsigned int curr_index;
};

}   // namespace dataloader
}   // namespace magmadnn