提交 3b076f98 authored 作者: John Salvatier's avatar John Salvatier

switched to using advinc package

上级 04120f85
...@@ -8,6 +8,7 @@ from itertools import izip ...@@ -8,6 +8,7 @@ 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
...@@ -7382,7 +7383,7 @@ class AdvancedIncSubtensor(Op): ...@@ -7382,7 +7383,7 @@ class AdvancedIncSubtensor(Op):
if not self.inplace: if not self.inplace:
a = inputs[0].copy() a = inputs[0].copy()
numpy.inplace_increment(a, tuple(inputs[2:]), inputs[1]) advinc.inplace_increment(a, tuple(inputs[2:]), inputs[1])
out[0] = a out[0] = a
out[0][inputs[2:]] = inputs[1] out[0][inputs[2:]] = inputs[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论