Add test

FluxML · Oct 4, 2024 · 9f59e80 · 9f59e80
1 parent fc37e47
commit 9f59e80
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/chainrules.jl b/test/chainrules.jl
@@ -419,6 +419,9 @@ end
     @test z2d_compiled.d === z2d_fallback.d
     @test z2d_compiled.c.a === z2d_fallback.c.a
     @test z2d_compiled.c.b === z2d_fallback.c.b
+
+    # empty arrays => NoTangent()
+    @test z2d(ones(1, 0), ones(16, 0)) === NoTangent()
 end
 
 @testset "ChainRules translation" begin