fix einsum issue (#67588) (#67825)

jzhang533 · web-flow · commit 8ce0de584c57 · 2024-09-06T21:01:22.000+08:00
Signed-off-by: Zhang Jun &lt;jzhang533@gmail.com&gt;
diff --git a/python/paddle/tensor/einsum.py b/python/paddle/tensor/einsum.py
@@ -754,7 +754,7 @@ def fake_shape(ori_label, label, op):
             "length of shape and length of label must be the same, but received %d != %d"
             % (len(op.shape), len(label))
         )
-        fakes = [s for i, (l, s) in enumerate(zip(label, op.shape)) if l != '.']
+        fakes = [s for i, (l, s) in enumerate(zip(label, op.shape))]
         fakes = list(map(abs, fakes))  # make -1 -> 1
         if '.' in ori_label:
             fakes.insert(ori_label.index('.'), 1)

Original file line number	Diff line number	Diff line change
`@@ -754,7 +754,7 @@ def fake_shape(ori_label, label, op):`
`754`	`754`	`"length of shape and length of label must be the same, but received %d != %d"`
`755`	`755`	`% (len(op.shape), len(label))`
`756`	`756`	`)`
`757`		`- fakes = [s for i, (l, s) in enumerate(zip(label, op.shape)) if l != '.']`
	`757`	`+ fakes = [s for i, (l, s) in enumerate(zip(label, op.shape))]`
`758`	`758`	`fakes = list(map(abs, fakes)) # make -1 -> 1`
`759`	`759`	`if '.' in ori_label:`
`760`	`760`	`fakes.insert(ori_label.index('.'), 1)`