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

Remove Hints and use instance checks in aesara.sandbox.linalg.ops

上级 2b12a455
from aesara.sandbox.linalg.ops import psd, spectral_radius_bound from aesara.sandbox.linalg.ops import spectral_radius_bound
from aesara.tensor.nlinalg import det, eig, eigh, matrix_inverse, trace
from aesara.tensor.slinalg import cholesky, eigvalsh, solve
...@@ -5,19 +5,11 @@ import aesara ...@@ -5,19 +5,11 @@ import aesara
from aesara import function from aesara import function
from aesara import tensor as aet from aesara import tensor as aet
from aesara.configdefaults import config from aesara.configdefaults import config
from aesara.sandbox.linalg.ops import inv_as_solve, spectral_radius_bound
# The one in comment are not tested...
from aesara.sandbox.linalg.ops import Cholesky # PSD_hint,; op class
from aesara.sandbox.linalg.ops import (
Solve,
inv_as_solve,
matrix_inverse,
solve,
spectral_radius_bound,
)
from aesara.tensor.elemwise import DimShuffle from aesara.tensor.elemwise import DimShuffle
from aesara.tensor.math import _allclose from aesara.tensor.math import _allclose
from aesara.tensor.nlinalg import MatrixInverse from aesara.tensor.nlinalg import MatrixInverse, matrix_inverse
from aesara.tensor.slinalg import Cholesky, Solve, solve
from aesara.tensor.type import dmatrix, matrix, vector from aesara.tensor.type import dmatrix, matrix, vector
from tests import unittest_tools as utt from tests import unittest_tools as utt
from tests.test_rop import break_op from tests.test_rop import break_op
...@@ -120,7 +112,7 @@ def test_spectral_radius_bound(): ...@@ -120,7 +112,7 @@ def test_spectral_radius_bound():
def test_transinv_to_invtrans(): def test_transinv_to_invtrans():
X = matrix("X") X = matrix("X")
Y = aesara.tensor.nlinalg.matrix_inverse(X) Y = matrix_inverse(X)
Z = Y.transpose() Z = Y.transpose()
f = aesara.function([X], Z) f = aesara.function([X], Z)
if config.mode != "FAST_COMPILE": if config.mode != "FAST_COMPILE":
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论