dataset.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <stdexcept>
#include <vector>
 
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
struct Sample {
    std::vector<Tensor> fields;
};
 
class Dataset {
public:
    virtual ~Dataset() = default;
 
    [[nodiscard]] virtual int64_t size() const = 0;
 
    [[nodiscard]] virtual Sample get(int64_t index) const = 0;
};
 
}  // namespace pulsatrix