提交 ae73ad8a authored 作者: Frederic Bastien's avatar Frederic Bastien

make a check more strict. This is usefull as AdvancedSubtensor give better message error.

上级 174b2093
......@@ -1169,7 +1169,9 @@ class _tensor_py_operators:
break
if advanced:
if len(args) == 1:
if len(args) == 1 and isinstance(args[0],
(list, TensorVariable,
theano.tensor.sharedvar.TensorSharedVariable)):
return advanced_subtensor1(self, *args)
else:
return AdvancedSubtensor(args)(self, *args)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论