提交 d3d6712f authored 作者: John Salvatier's avatar John Salvatier

fixed rebase errors

上级 37848106
...@@ -1761,12 +1761,18 @@ class _tensor_py_operators: ...@@ -1761,12 +1761,18 @@ class _tensor_py_operators:
axis = i axis = i
if advanced: if advanced:
if (axis is not None
and numpy.all(a == slice(None) for a in args[:axis])
and numpy.all(a == slice(None) for a in args[axis + 1:]) and numpy.all(a == slice(None) for a in args[axis + 1:])
and isinstance(args[axis], ( and isinstance(args[axis], (
numpy.ndarray, numpy.ndarray,
theano.tensor.sharedvar.TensorSharedVariable)) list,
and as_tensor_variable(args[0]).type.broadcastable == (False,) ): TensorVariable,
return AdvancedSubtensor()(self, *args) TensorConstant,
theano.tensor.sharedvar.TensorSharedVariable))):
return self.take(arg, axis)
else:
return AdvancedSubtensor()(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
...@@ -7438,7 +7444,6 @@ class AdvancedIncSubtensor(Op): ...@@ -7438,7 +7444,6 @@ class AdvancedIncSubtensor(Op):
increment(out[0], tuple(inputs[2:]), inputs[1]) increment(out[0], tuple(inputs[2:]), inputs[1])
out[0][inputs[2:]] = inputs[1] out[0][inputs[2:]] = inputs[1]
else:
if (numpy.__version__ <= '1.6.1' and if (numpy.__version__ <= '1.6.1' and
out[0].size != numpy.uint32(out[0].size)): out[0].size != numpy.uint32(out[0].size)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论