提交 590d2601 authored 作者: Frederic's avatar Frederic

Don't always instantiate new object.

上级 280cb21b
...@@ -1847,6 +1847,7 @@ class AdvancedSubtensor(Op): ...@@ -1847,6 +1847,7 @@ class AdvancedSubtensor(Op):
return [advanced_inc_subtensor(theano.tensor.zeros_like(x), gz, return [advanced_inc_subtensor(theano.tensor.zeros_like(x), gz,
*rest)] + \ *rest)] + \
[DisconnectedType()()] * len(rest) [DisconnectedType()()] * len(rest)
advanced_subtensor = AdvancedSubtensor()
class AdvancedIncSubtensor(Op): class AdvancedIncSubtensor(Op):
......
...@@ -378,7 +378,7 @@ class _tensor_py_operators: ...@@ -378,7 +378,7 @@ class _tensor_py_operators:
theano.tensor.sharedvar.TensorSharedVariable))): theano.tensor.sharedvar.TensorSharedVariable))):
return self.take(arg, axis) return self.take(arg, axis)
else: else:
return theano.tensor.subtensor.AdvancedSubtensor()(self, *args) return theano.tensor.subtensor.advanced_subtensor(self, *args)
else: else:
if numpy.newaxis in args: if numpy.newaxis in args:
# None (aka np.newaxis) in numpy indexing means to add a # None (aka np.newaxis) in numpy indexing means to add a
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论