Remove wrong transpose (#161)

* Remove wrong transpose

* Fix

Author: gdalle

Project.toml

@@ -1,7 +1,7 @@
 name = "ImplicitDifferentiation"
 uuid = "57b37032-215b-411a-8a7c-41a003a55207"
 authors = ["Guillaume Dalle", "Mohamed Tarek and contributors"]
-version = "0.6.2"
+version = "0.6.3"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

ext/ImplicitDifferentiationForwardDiffExt.jl

@@ -30,10 +30,10 @@ function (implicit::ImplicitFunction)(
         kwargs...,
     )
 
-    dX = mapreduce(hcat, 1:N) do k
+    dX = map(1:N) do k
         partials.(x_and_dx, k)
     end
-    dC = mapreduce(hcat, eachcol(dX)) do dₖx
+    dC = mapreduce(hcat, dX) do dₖx
         B * dₖx
     end
     dY = implicit.linear_solver(A, -dC)

src/operators.jl

@@ -164,7 +164,7 @@ function build_B(
             typeof(x),
         )
     else
-        B = transpose(jacobian(cond_x, back_x, x, contexts...))
+        B = jacobian(cond_x, back_x, x, contexts...)
     end
     return B
 end