体验新版 GitCode,发现更多精彩内容 >>
#pragma once namespace mgb::imperative::python { PyObject* make_shape_tuple(PyObject* self, PyObject* const* args, size_t nargs); PyObject* getitem_cpp(PyObject* self, PyObject* const* args, size_t nargs); PyObject* setitem_cpp(PyObject* self, PyObject* const* args, size_t nargs);
PyObject* split_cpp(PyObject* self, PyObject* const* args, size_t nargs);
PyObject* expand_dims_cpp(PyObject* self, PyObject* const* args, size_t nargs);
PyObject* squeeze_cpp(PyObject* self, PyObject* const* args, size_t nargs);
PyObject* transpose_cpp(PyObject* self, PyObject* const* args, size_t nargs);
PyObject* broadcast_cpp(PyObject* self, PyObject* const* args, size_t nargs); PyObject* reshape_cpp(PyObject* self, PyObject* const* args, size_t nargs);
} // namespace mgb::imperative::python