|
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