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

added TensorConstant to list of AdvancedSubtensor1 triggers

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