文件 python_callbacks.h

struct PyCallbackIOWriter : public faiss::IOWriter 

公共函数

explicit PyCallbackIOWriter(PyObject *callback, size_t bs = 1024 * 1024): Callback: Python 函数，接受一个字节对象并返回成功写入的字节数。

virtual size_t operator()(const void *ptrv, size_t size, size_t nitems) override

~PyCallbackIOWriter() override

公共成员

PyObject *callback

size_t bs

struct PyCallbackIOReader : public faiss::IOReader 

公共函数

explicit PyCallbackIOReader(PyObject *callback, size_t bs = 1024 * 1024): Callback: 接受大小参数并返回带有结果读取的字节对象的 Python 函数。

virtual size_t operator()(void *ptrv, size_t size, size_t nitems) override

~PyCallbackIOReader() override

公共成员

PyObject *callback

size_t bs

struct PyCallbackIDSelector : public faiss::IDSelector 

公共函数

explicit PyCallbackIDSelector(PyObject *callback)

virtual bool is_member(faiss::idx_t id) const override

~PyCallbackIDSelector() override

公共成员

PyObject *callback