/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/cpp_bindings/dataloader_py.cpp
20 строк
567 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
#include <pybind11/pybind11.h> #include <pybind11/stl.h> #include <adept/data/dataloader.hpp> namespace py = pybind11; using namespace adept; void bind_dataloader(py::module& m) { py::class_<DataLoader>(m, "DataLoader") .def(py::init<Dataset*, index_t>()) .def( "__iter__", [](DataLoader& data_loader) { return py::make_iterator(data_loader.begin(), data_loader.end()); }, py::keep_alive<0, 1>() /* Essential: keep object alive while iterator exists */) .def("__len__", &DataLoader::size); }