提交 95adecd6 authored 作者: Shawn Tan's avatar Shawn Tan

Re-included scipy check from original library.

上级 f9202f68
...@@ -24,7 +24,9 @@ from theano.tensor.nlinalg import (MatrixInverse, ...@@ -24,7 +24,9 @@ from theano.tensor.nlinalg import (MatrixInverse,
from theano.tensor.slinalg import (Cholesky, from theano.tensor.slinalg import (Cholesky,
cholesky, cholesky,
Solve, Solve,
solve) solve,
imported_scipy)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -244,6 +246,8 @@ def transinv_to_invtrans(node): ...@@ -244,6 +246,8 @@ def transinv_to_invtrans(node):
@register_stabilize @register_stabilize
@local_optimizer([Dot, Dot22]) @local_optimizer([Dot, Dot22])
def inv_as_solve(node): def inv_as_solve(node):
if not imported_scipy:
return False
if isinstance(node.op, (Dot, Dot22)): if isinstance(node.op, (Dot, Dot22)):
l, r = node.inputs l, r = node.inputs
if l.owner and l.owner.op == matrix_inverse: if l.owner and l.owner.op == matrix_inverse:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论