more tests on BroadCast

Author: nantonel

src/calculus/Jacobian.jl

@@ -71,6 +71,8 @@ Jacobian(S::Sum{M,N,K,C,D},x::D) where {M,N,K,C,D} =
 Sum(([Jacobian(a,x) for a in S.A]...),S.bufC,S.bufD,M,N)
 #Jacobian of Transpose
 Jacobian(T::Transpose{A}, x::AbstractArray) where {A <: AbstractOperator} = T 
+#Jacobian of BroadCast
+Jacobian(B::A, x::AbstractArray) where {A <: BroadCast} = BroadCast(Jacobian(B.A,x),B.dim_out) 
 
 # Properties
 

test/test_linear_operators_calculus.jl

@@ -740,6 +740,16 @@ y2 = zeros(dim_out)
 y2 .= x1
 @test vecnorm(y1-y2) <= 1e-12
 
+m, n, l  = 1, 5, 8
+dim_out = (m, n, l)
+opA1 = Scale(2.4,Eye(m))
+opR = BroadCast(opA1, dim_out)
+x1 = randn(m)
+y1 = test_op(opR, x1, randn(dim_out), verb)
+y2 = zeros(dim_out)
+y2 .= 2.4*x1
+@test vecnorm(y1-y2) <= 1e-12
+
 @test is_null(opR)             == is_null(opA1)            
 @test is_eye(opR)              == false             
 @test is_diagonal(opR)         == false 

test/test_nonlinear_operators_calculus.jl

@@ -143,6 +143,29 @@ y, grad = test_NLop(op,x,r,verb)
 Y = reshape(opS*x,2,2)
 @test vecnorm(Y-y) <1e-8
 
+###testing BroadCast
+n,l = 4,7
+x = randn(n)
+r = randn(n,l)
+opS = Sigmoid(Float64,(n,),2)
+op = BroadCast(opS,(n,l))
+
+y, grad = test_NLop(op,x,r,verb)
+
+Y = (opS*x).*ones(n,l)
+@test vecnorm(Y-y) <1e-8
+
+n,l = 1,7
+x = randn(n)
+r = randn(n,l)
+opS = Sigmoid(Float64,(n,),2)
+op = BroadCast(opS,(n,l))
+
+y, grad = test_NLop(op,x,r,verb)
+
+Y = (opS*x).*ones(n,l)
+@test vecnorm(Y-y) <1e-8
+
 ##testing Sum
 m = 5
 x = randn(m)