提交 c803c67e authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Load the sparse shared constructor support at package-level

上级 b4442683
...@@ -176,6 +176,7 @@ def get_scalar_constant_value(v): ...@@ -176,6 +176,7 @@ def get_scalar_constant_value(v):
# isort: off # isort: off
import aesara.tensor.random.var import aesara.tensor.random.var
import aesara.sparse
from aesara.scan import checkpoints from aesara.scan import checkpoints
from aesara.scan.basic import scan from aesara.scan.basic import scan
from aesara.scan.views import foldl, foldr, map, reduce from aesara.scan.views import foldl, foldr, map, reduce
......
import numpy as np
import pytest
import aesara
from aesara.compile import SharedVariable
sp = pytest.importorskip("scipy", minversion="0.7.0")
def test_shared_basic():
x = aesara.shared(sp.sparse.csr_matrix(np.eye(100)), name="blah", borrow=True)
assert isinstance(x, SharedVariable)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论