提交 480f45dc authored 作者: Caglar's avatar Caglar

Added linalg import back.

上级 b71bd504
...@@ -1513,6 +1513,7 @@ def local_gpu_extract_diagonal(node): ...@@ -1513,6 +1513,7 @@ def local_gpu_extract_diagonal(node):
""" """
global linalg global linalg
if linalg is None: if linalg is None:
from theano.sandbox import linalg
linalg = theano.sandbox.linalg linalg = theano.sandbox.linalg
if (isinstance(node.op, linalg.ops.ExtractDiag) and if (isinstance(node.op, linalg.ops.ExtractDiag) and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论