Use fill(x[], ()) instead

TuringLang · Oct 25, 2024 · 8c9cc71 · 8c9cc71
1 parent db391a2
commit 8c9cc71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -290,7 +290,7 @@ function (f::ReshapeTransform)(x)
         # Specially handle the case where x is a singleton array, see
         # https://github.com/JuliaDiff/ReverseDiff.jl/issues/265 and
         # https://github.com/TuringLang/DynamicPPL.jl/issues/698
-        return x[]
+        return fill(x[], ())
     else
         # The call to `tovec` is only needed in case `x` is a scalar.
         return reshape(tovec(x), f.output_size)