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

View File

@ -486,7 +486,7 @@ def einsum(string, *args):
def einsum_outshape(einsum_expr, 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('->')
if s[-1]=='':
return ()