From 7db591ef0a17ca7c05adb4c53ed13a9bef7b562f Mon Sep 17 00:00:00 2001
From: guoye <498731903@qq.com>
Date: Thu, 28 May 2020 15:08:37 +0800
Subject: [PATCH] sqr()

---
 python/jittor/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py
index a1bdc29f..40cbdba9 100644
--- a/python/jittor/__init__.py
+++ b/python/jittor/__init__.py
@@ -186,7 +186,7 @@ def norm(x, k, dim):
     if k==1:
         return x.abs().sum(dim)
     if k==2:
-        return (x**2).sum(dim).sqrt()
+        return x.sqr().sum(dim).sqrt()
 Var.norm = norm
 
 origin_reshape = reshape