Add test for Stacked with vectors

TuringLang · Jan 14, 2025 · 4946dbf · 4946dbf
1 parent 73f3459
commit 4946dbf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/ad/stacked.jl b/test/ad/stacked.jl
@@ -24,4 +24,15 @@
     test_ad(y) do y
         sum(transform(binv, y))
     end
+
+    bvec = Stacked([b1, b2], [1:4, 5:5])
+    bvec_inv = inverse(bvec)
+
+    test_ad(y) do x
+        sum(transform(bvec, binv(x)))
+    end
+
+    test_ad(y) do y
+        sum(transform(bvec_inv, y))
+    end
 end