From 84f990a05175a4380db9c605b717e257ea9ca70b Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 16 Mar 2021 11:29:06 +0800
Subject: [PATCH] fix(mge/tensor): fix `__repr__` for Tensor's subclass

GitOrigin-RevId: 91b58e6dcaacc40c3dac6d3689ed5f5a7a9be7dd
---
 imperative/python/megengine/tensor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/megengine/tensor.py b/imperative/python/megengine/tensor.py
index 85d98781..de4afd33 100644
--- a/imperative/python/megengine/tensor.py
+++ b/imperative/python/megengine/tensor.py
@@ -153,7 +153,7 @@ class Tensor(_Tensor, ArrayMethodMixin):
         super()._reset(other)
 
     def __repr__(self):
-        piece = "Tensor("
+        piece = "{}(".format(self.__class__.__name__)
         with np.printoptions(precision=4, suppress=True):
             piece += "{}".format(str(self.numpy()))
         if self.dtype != np.float32:
-- 
GitLab