提交 dc91b431 authored 作者: James Bergstra's avatar James Bergstra

added TensorConstant to list of AdvancedSubtensor1 triggers

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