From d3c7c56317ddb7da428d102416acb1fc8f5c13fc Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 29 Nov 2021 18:23:41 +0800
Subject: [PATCH] fix(traced_module): fix tensor node init

GitOrigin-RevId: bdd8259ce7db99f3da35c7748b3b5a5c2d807a54
---
 imperative/python/megengine/traced_module/node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/megengine/traced_module/node.py b/imperative/python/megengine/traced_module/node.py
index c16ecdb0..4bfff4af 100644
--- a/imperative/python/megengine/traced_module/node.py
+++ b/imperative/python/megengine/traced_module/node.py
@@ -231,7 +231,7 @@ class TensorNode(Node):
     ):
         super().__init__(expr, name, qualname)
         self._shape = shape
-        self._dtype = shape
+        self._dtype = dtype
         self._qparams = qparams
 
     def __getstate__(self):