|
162 | 162 | @testset "kron" begin
|
163 | 163 | @testset "AbstractVecOrMat{$T}" for T in (Float64, ComplexF64)
|
164 | 164 | @testset "frule" begin
|
165 |
| - test_frule(kron, randn(T, 3), randn(T, 3)) |
166 |
| - test_frule(kron, randn(T, 3, 2), randn(T, 3)) |
167 |
| - test_frule(kron, randn(T, 3), randn(T, 3, 4)) |
168 |
| - test_frule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 165 | + test_frule(kron, randn(T, 2), randn(T, 3)) |
| 166 | + test_frule(kron, randn(T, 2, 3), randn(T, 5)) |
| 167 | + test_frule(kron, randn(T, 2), randn(T, 3, 5)) |
| 168 | + test_frule(kron, randn(T, 2, 3), randn(T, 5, 7)) |
169 | 169 | end
|
170 | 170 | @testset "rrule" begin
|
171 |
| - test_rrule(kron, randn(T, 3), randn(T, 3)) |
172 |
| - test_rrule(kron, randn(T, 3, 2), randn(T, 3)) |
173 |
| - test_rrule(kron, randn(T, 3), randn(T, 3, 4)) |
174 |
| - test_rrule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 171 | + test_rrule(kron, randn(T, 2), randn(T, 3)) |
| 172 | + test_rrule(kron, randn(T, 2, 3), randn(T, 5)) |
| 173 | + test_rrule(kron, randn(T, 2), randn(T, 3, 5)) |
| 174 | + test_rrule(kron, randn(T, 2, 3), randn(T, 5, 7)) |
175 | 175 | end
|
176 | 176 | end
|
177 | 177 | end
|
|
0 commit comments