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

fix einsum space interpret error.

This commit is contained in:
Exusial 2022-08-27 22:25:00 +08:00
parent b6c9421a9a
commit 6339f62f56
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/linalg.py
View File

@ -486,7 +486,7 @@ def einsum(string, *args):
def einsum_outshape(einsum_expr, inputs): def einsum_outshape(einsum_expr, inputs):
shps = np_cpu.concatenate([in_.shape for in_ in inputs]) shps = np_cpu.concatenate([in_.shape for in_ in inputs])
p = einsum_expr.split(',') p = einsum_expr.replace(" ", "").split(',')
s = p[:-1] + p[-1].split('->') s = p[:-1] + p[-1].split('->')
if s[-1]=='': if s[-1]=='':
return () return ()