sparsevec test

Author: CarloLucibello

src/rulesets/SparseArrays/sparsematrix.jl

@@ -44,7 +44,7 @@ function rrule(::typeof(findnz), v::AbstractSparseVector)
     function findnz_pullback(Δ)
         _, V̄ = unthunk(Δ)
         V̄ isa AbstractZero && return (NoTangent(), V̄)
-        return NoTangent(), sparse(I, V̄, n)
+        return NoTangent(), sparsevec(I, V̄, n)
     end
 
     return (I, V), findnz_pullback

test/rulesets/SparseArrays/sparsematrix.jl

@@ -25,4 +25,11 @@ end
     I, J, V = findnz(A)
     V̄ = rand!(similar(V))
     test_rrule(findnz, A ⊢ dA, output_tangent=(zeros(length(I)), zeros(length(J)), V̄))
+
+    v = sprand(5, 0.5)
+    dv = similar(v)
+    rand!(dv.nzval)
+    I, V = findnz(v)
+    V̄ = rand!(similar(V))
+    test_rrule(findnz, v ⊢ dv, output_tangent=(zeros(length(I)), V̄))
 end