|
161 | 161 | end |
162 | 162 | @testset "kron" begin |
163 | 163 | @testset "AbstractVecOrMat{$T}" for T in (Float64, ComplexF64) |
164 | | - test_frule(kron, randn(T, 3), randn(T, 3)) |
165 | | - test_frule(kron, randn(T, 3, 2), randn(T, 3)) |
166 | | - test_frule(kron, randn(T, 3), randn(T, 3, 4)) |
167 | | - test_frule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 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)) |
| 169 | + end |
| 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)) |
| 175 | + end |
168 | 176 | end |
169 | 177 | end |
170 | 178 | end |
0 commit comments