lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

sqr()

This commit is contained in:
guoye 2020-05-28 15:08:37 +08:00
parent 5bcfe0db60
commit 7db591ef0a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -186,7 +186,7 @@ def norm(x, k, dim):
if k==1: if k==1:
return x.abs().sum(dim) return x.abs().sum(dim)
if k==2: if k==2:
return (x**2).sum(dim).sqrt() return x.sqr().sum(dim).sqrt()
Var.norm = norm Var.norm = norm
origin_reshape = reshape origin_reshape = reshape