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

fix cumprod

This commit is contained in:
cxjyxx_me 2021-09-02 04:10:17 -04:00
parent 8a2e7a1881
commit 546860e19e
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/misc.py
View File

@ -718,7 +718,7 @@ def cumsum(x, dim=None):
jt.Var.cumsum = cumsum
def cumprod(x,dim=0):
def cumprod(x,dim=None):
x = jt.log(x)
x = cumsum(x,dim=dim)
return jt.exp(x)