提交 01354337 编写于 作者: M Megvii Engine Team

fix(mge/autodiff): fix incorrect handling of tuple dy

GitOrigin-RevId: beca8e37117d1b1f24d463f39903a1a74a62d167
上级 5868d1fe
......@@ -262,7 +262,7 @@ class GradManager:
if dy is None:
dys = [ones_like(y) for y in ys]
elif isinstance(dy, (tuple, list)):
dys = ys
dys = dy
else:
dys = [dy]
try:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册