Merge pull request #238 from dfdx/dfdx/fix-rand-tangent

Add rand_tangent() for Broadcasted

Author: oxinabox

Project.toml

@@ -1,6 +1,6 @@
 name = "ChainRulesTestUtils"
 uuid = "cdddcdb0-9152-4a09-a978-84456f9df70a"
-version = "1.5.1"
+version = "1.5.2"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/rand_tangent.jl

@@ -24,6 +24,8 @@ rand_tangent(rng::AbstractRNG, ::BigFloat) = round(big(9 * randn(rng)), digits=5
 
 rand_tangent(rng::AbstractRNG, x::AbstractArray) = ProjectTo(x)(rand_tangent.(Ref(rng), x))
 
+rand_tangent(rng::AbstractRNG, x::Broadcast.Broadcasted) = rand_tangent(Broadcast.materialize(x))
+
 function rand_tangent(rng::AbstractRNG, x::T) where {T}
     if !isstructtype(T)
         throw(ArgumentError("Non-struct types are not supported by this fallback."))

test/rand_tangent.jl

@@ -17,7 +17,7 @@ struct Bar
         (4, NoTangent),
         (FiniteDifferences, NoTangent),  # Module object
         # Types (not instances of type)
-        (Bar, NoTangent),  
+        (Bar, NoTangent),
         (Union{Int, Bar}, NoTangent),
         (Union{Int, Bar}, NoTangent),
         (Vector, NoTangent),
@@ -42,15 +42,18 @@ struct Bar
         # Co-Arrays
         (randn(5)', Adjoint{Float64, Vector{Float64}}),  # row-vector: special
         (randn(5, 4)', Matrix{Float64}),                 # matrix: generic dense
-        
+
         (transpose(randn(5)), Transpose{Float64, Vector{Float64}}),  # row-vector: special
         (transpose(randn(5, 4)), Matrix{Float64}),                   # matrix: generic dense
-        
+
         # AbstactArrays of non-perturbable types
         (1:10, NoTangent),
         (1:2:10, NoTangent),
         ([false, true], NoTangent),
 
+        # Broadcasted.
+        (Broadcast.broadcasted(sin, randn(5, 4)), Matrix{Float64}),
+
         # Tuples.
         ((4.0, ), Tangent{Tuple{Float64}}),
         ((5.0, randn(3)), Tangent{Tuple{Float64, Vector{Float64}}}),