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 1
1
import numpy as np
2
2
import pytest
3
3
4
- import pytensor .tensor as pt
5
4
from pytensor .configdefaults import config
6
5
from pytensor .graph .fg import FunctionGraph
7
6
from pytensor .tensor import slinalg as pt_slinalg
@@ -128,18 +127,3 @@ def test_pytorch_block_diag():
128
127
np .random .normal (size = (4 , 4 )).astype (config .floatX ),
129
128
],
130
129
)
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