File tree Expand file tree Collapse file tree 1 file changed +0
-16
lines changed
Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Original file line number Diff line number Diff line change 11import numpy as np
22import pytest
33
4- import pytensor .tensor as pt
54from pytensor .configdefaults import config
65from pytensor .graph .fg import FunctionGraph
76from pytensor .tensor import slinalg as pt_slinalg
@@ -128,18 +127,3 @@ def test_pytorch_block_diag():
128127 np .random .normal (size = (4 , 4 )).astype (config .floatX ),
129128 ],
130129 )
131-
132-
133- @pytest .mark .xfail (reason = "Blockwise not implemented" )
134- def test_pytorch_block_diag_blockwise ():
135- A = pt .tensor3 ("A" )
136- B = pt .tensor3 ("B" )
137- out = pt_slinalg .block_diag (A , B )
138- out_fg = FunctionGraph ([A , B ], [out ])
139- compare_pytorch_and_py (
140- out_fg ,
141- [
142- np .random .normal (size = (5 , 5 , 5 )).astype (config .floatX ),
143- np .random .normal (size = (5 , 3 , 3 )).astype (config .floatX ),
144- ],
145- )
You can’t perform that action at this time.
0 commit comments