提交 7e724eea authored 作者: Saizheng Zhang's avatar Saizheng Zhang

change to node.op in tensor_inc_subtensor

上级 4145a327
...@@ -1916,7 +1916,7 @@ def local_subtensor_inc_subtensor(node): ...@@ -1916,7 +1916,7 @@ def local_subtensor_inc_subtensor(node):
return [x.owner.inputs[1]] return [x.owner.inputs[1]]
# else y is broadcastable, return alloc of broadcastable y # else y is broadcastable, return alloc of broadcastable y
else: else:
x_subtensor = Subtensor(node.op.idx_list)(x.owner.inputs[0], *x.owner.inputs[2:]) x_subtensor = node.op(x.owner.inputs[0], *x.owner.inputs[2:])
return [T.alloc(x.owner.inputs[1], *x_subtensor.shape)] return [T.alloc(x.owner.inputs[1], *x_subtensor.shape)]
else: else:
return return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论