提交 325a73a2 authored 作者: John Salvatier's avatar John Salvatier

added check for inplace_increment

上级 98b979fe
...@@ -8,7 +8,6 @@ from itertools import izip ...@@ -8,7 +8,6 @@ from itertools import izip
from textwrap import dedent from textwrap import dedent
import numpy import numpy
import advinc
#from copy import copy as python_copy #from copy import copy as python_copy
import theano import theano
...@@ -7399,11 +7398,21 @@ class AdvancedIncSubtensor(Op): ...@@ -7399,11 +7398,21 @@ class AdvancedIncSubtensor(Op):
out, = out_ out, = out_
if not self.inplace: if not self.inplace:
out[0] = inputs[0].copy()
a = inputs[0].copy() else:
advinc.inplace_increment(a, tuple(inputs[2:]), inputs[1]) raise NotImplementedError('In place computation is not'
out[0] = a ' implemented')
if self.set_instead_of_inc:
out[0][inputs[2:]] = inputs[1]
else:
try :
increment = gof.cutils_ext.inplace_increment
except:
raise NotImplementedError("Couldn't find
inplace_increment, update numpy.")
increment(out[0], tuple(inputs[2:]), inputs[1])
out[0][inputs[2:]] = inputs[1] out[0][inputs[2:]] = inputs[1]
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论