提交 71dea2cf authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix problem with targ.ndim.

上级 393940fd
......@@ -160,9 +160,11 @@ def alpha_merge(cls, alpha_in, beta_in):
targ = find_node(node.inputs[1], cls)
if targ is None:
return
lr = grab_cpu_scalar(node.inputs[0], nd=targ.ndim)
lr = grab_cpu_scalar(node.inputs[0],
nd=targ.outputs[0].ndim)
else:
lr = grab_cpu_scalar(node.inputs[1], nd=targ.ndim)
lr = grab_cpu_scalar(node.inputs[1],
nd=targ.outputs[0].ndim)
if lr is None or lr.dtype != targ.outputs[0].dtype:
return None
inputs = list(targ.inputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论