From c6bbc478c7f1fe526c8d0b7e4e00872223774f05 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 31 Mar 2021 18:08:12 +0800
Subject: [PATCH] feat(imperative/utils): add __repr__ for network node

GitOrigin-RevId: fe5b59749fb98fc9fdca61421ff6e6d657472fe4
---
 imperative/python/megengine/utils/network_node.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/imperative/python/megengine/utils/network_node.py b/imperative/python/megengine/utils/network_node.py
index 5e1f1af33..155b05796 100644
--- a/imperative/python/megengine/utils/network_node.py
+++ b/imperative/python/megengine/utils/network_node.py
@@ -99,6 +99,9 @@ class OpNode(NetworkNode):
     def add_out_var(self, x):
         self.outputs.append(x)
 
+    def __repr__(self):
+        return "%s{%s}" % (self.name, self.type)
+
 
 def str_to_mge_class(classname):
     # TODO: use megbrain C++ RTTI to replace type string
@@ -230,11 +233,17 @@ class Elemwise(OpNode):
     type = "Elemwise"
     opdef = builtin.Elemwise
 
+    def __repr__(self):
+        return "%s{Elemwise:%s}" % (self.name, self.params["mode"])
+
 
 class ElemwiseMultiType(OpNode):
     type = "ElemwiseMultiType"
     opdef = builtin.ElemwiseMultiType
 
+    def __repr__(self):
+        return "%s{ElemwiseMultiType:%s}" % (self.name, self.params["mode"])
+
     @classmethod
     def load(cls, opr):
         obj = super(ElemwiseMultiType, cls).load(opr)
-- 
GitLab