/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/cpp_bindings/mnistdataset_py.cpp
17 строк
533 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/mnistdataset.hpp> namespace py = pybind11; using namespace adept; void bind_mnistdataset(py::module& m) { py::class_<MNISTDataset, Dataset>(m, "MNISTDataset") .def(py::init<const std::string&, const std::string&, dtype_t, device_t, bool>()) .def("size", &MNISTDataset::size) .def("__len__", [](MNISTDataset& self) { return self.size(); }) .def("item", &MNISTDataset::item) .def("item_size", &MNISTDataset::item_size); }