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

Added linalg import back.

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