Skip to content
Snippets Groups Projects
Commit f6f1c60c authored by Maxence Naud's avatar Maxence Naud
Browse files

[Doc] pybind_GraphView

parent 35daaed6
No related branches found
No related tags found
No related merge requests found
...@@ -63,10 +63,26 @@ void init_GraphView(py::module& m) { ...@@ -63,10 +63,26 @@ void init_GraphView(py::module& m) {
IOIndex_t)) & IOIndex_t)) &
GraphView::addChild, GraphView::addChild,
py::arg("toOtherNode"), py::arg("fromOutNode") = nullptr, py::arg("toOtherNode"), py::arg("fromOutNode") = nullptr,
py::arg("fromTensor") = 0U, py::arg("toTensor") = gk_IODefaultIndex) py::arg("fromTensor") = 0U, py::arg("toTensor") = gk_IODefaultIndex,
R"mydelimiter(
Include a Node to the current GraphView object.
:param other_node: Node to add
:type oth_Node: Node
:param includeLearnableParameter: include non-data inputs, like weights and biases. Default True.
:type includeLearnableParameter
)mydelimiter")
.def("replace_with", &GraphView::replaceWith, py::arg("new_nodes"),
R"mydelimiter(
Replace the current GraphView with the set of given Nodes if possible.
.def("replace_with", &GraphView::replaceWith, py::arg("new_nodes")) :param new_nodes: Nodes with connections already taken care of.
:type new_nodes: Node
:return: Whether any replacement has been made.
:rtype: bool
)mydelimiter")
.def("get_nodes", &GraphView::getNodes) .def("get_nodes", &GraphView::getNodes)
.def("get_node", &GraphView::getNode, py::arg("node_name")) .def("get_node", &GraphView::getNode, py::arg("node_name"))
.def("forward_dims", &GraphView::forwardDims) .def("forward_dims", &GraphView::forwardDims)
...@@ -87,4 +103,4 @@ void init_GraphView(py::module& m) { ...@@ -87,4 +103,4 @@ void init_GraphView(py::module& m) {
// }) // })
; ;
} }
} // namespace Aidge } // namespace Aidge
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment