pulsatrix/api/mnist__loader_8hpp_source.html

#pragma once


#include <cstdint>

#include <string>

#include <vector>


#include "pulsatrix/device_backend.hpp"

#include "pulsatrix/tensor.hpp"


namespace pulsatrix {


struct MnistDataset {

    std::vector<Tensor> images;

    std::vector<int64_t> labels;

};


class MnistIdxLoader {

public:

    [[nodiscard]] static MnistDataset Load(const std::string& images_path, const std::string& labels_path,

                                            DeviceBackend* backend, int64_t max_count = -1);

};


}  // namespace pulsatrix

pulsatrix::DeviceBackend
Vendor-agnostic compute/memory backend. CPUBackend, CUDABackend (Phase 1.5), and HIPBackend (Phase 1....
Definition device_backend.hpp:219

pulsatrix::MnistIdxLoader
Reads MNIST's original IDX-format files directly – no format conversion, no generic Dataset abstracti...
Definition mnist_loader.hpp:34

pulsatrix::MnistIdxLoader::Load
static MnistDataset Load(const std::string &images_path, const std::string &labels_path, DeviceBackend *backend, int64_t max_count=-1)
Loads one images/labels IDX file pair.

device_backend.hpp
Abstract interface isolating vendor-specific memory/compute operations from Tensor/ComputationGraph.

pulsatrix
Definition acquisition_functions.hpp:16

pulsatrix::MnistDataset
One IDX file pair's contents: parallel images/labels, same length.
Definition mnist_loader.hpp:18

pulsatrix::MnistDataset::images
std::vector< Tensor > images
Shape (1, 28, 28), pixel values normalized to [0,1].
Definition mnist_loader.hpp:20

pulsatrix::MnistDataset::labels
std::vector< int64_t > labels
0-9 class index, one per image, same order/length as images.
Definition mnist_loader.hpp:22

tensor.hpp
N-dimensional tensor – owns a buffer via DeviceBackend*, RAII (Rule of Five).