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

make sure advanced indexing only triggers 1d indexing if it's 1d

上级 676308df
......@@ -1764,6 +1764,8 @@ class _tensor_py_operators:
and numpy.all(a == slice(None) for a in args[axis + 1:])
and isinstance(args[axis], (
numpy.ndarray,
theano.tensor.sharedvar.TensorSharedVariable))
and as_tensor_variable(args[0]).type.broadcastable == (False,) ):
return AdvancedSubtensor()(self, *args)
else:
if numpy.newaxis in args:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论