Add inputsSize to Python IR, like outputsSize (#46779)

Summary: Pull Request resolved: #46779 Test Plan: Used it in some notebooks. Reviewed By: suo Differential Revision: D24574005 Pulled By: dreiss fbshipit-source-id: 78ba7a2bdb859fef5633212b73c7a3eb2cfbc380
pytorch · Oct 28, 2020 · 23bce17 · 23bce17
1 parent 179d2b2
commit 23bce17
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/csrc/jit/python/python_ir.cpp b/torch/csrc/jit/python/python_ir.cpp
@@ -493,6 +493,7 @@ void initPythonIRBindings(PyObject* module_) {
           })
       .def("sourceRange", [](Node& n) { return n.sourceRange().str(); })
       .def("hasMultipleOutputs", [](Node& n) { return n.outputs().size() > 1; })
+      .def("inputsSize", [](Node& n) { return n.inputs().size(); })
       .def("outputsSize", [](Node& n) { return n.outputs().size(); })
       .NS(kind)
       .def("inputsAt", [](Node& n, size_t i) { return n.inputs().at(i); })