diff --git a/python_binding/operator/pybind_Slice.cpp b/python_binding/operator/pybind_Slice.cpp index a024076e1dde4da6608ddc43861ffa20bfe217a8..b87cc8da4874c666de21a6e798a66e3c7fad9c10 100644 --- a/python_binding/operator/pybind_Slice.cpp +++ b/python_binding/operator/pybind_Slice.cpp @@ -23,10 +23,12 @@ void init_Slice(py::module& m) { py::class_<Slice_Op, std::shared_ptr<Slice_Op>, OperatorTensor>(m, "SliceOp", py::multiple_inheritance()) .def(py::init<const std::vector<std::int64_t>&, const std::vector<std::int64_t>&, - const std::vector<std::int8_t>&>(), + const std::vector<std::int8_t>&, + const std::vector<std::int64_t>&>(), py::arg("starts"), py::arg("ends"), - py::arg("axes")) + py::arg("axes"), + py::arg("steps")) .def_static("get_inputs_name", &Slice_Op::getInputsName) .def_static("get_outputs_name", &Slice_Op::getOutputsName); declare_registrable<Slice_Op>(m, "SliceOp");