fix(traced_module): fix tensor node init

GitOrigin-RevId: bdd8259ce7db99f3da35c7748b3b5a5c2d807a54

fix(traced_module): fix tensor node init
GitOrigin-RevId: bdd8259ce7db99f3da35c7748b3b5a5c2d807a54
fefd4f09 · Megvii Engine Team · “wenjuan” · 605836a7 · fefd4f09
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

imperative/python/megengine/traced_module/node.py imperative/python/megengine/traced_module/node.py +1 -1

未找到文件。
--- 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):