TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update tests

Browse Source
This commit is contained in:
Ali Hamdi 2017-06-08 20:59:46 +02:00
parent c350bfb672
commit 6a3f02b8e8
1 changed files with 19 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
test/backend/tensorflow.jl
View File

@ -25,28 +25,26 @@ end
@testset "Ops" begin @testset "Ops" begin
error_margin = 1e-4 @testset "svd" begin
#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# svd
A = convert(Array{Float32},randn(5,5)) A = convert(Array{Float32},randn(5,5))
@net f(x) = svd(x) @net f(x) = svd(x)
m = tf(f) m = tf(f)
u,s,v = m(A) u,s,v = m(A)
@test maximum(abs.(u*diagm(s)*transpose(v) - A)) < error_margin @test A ≈ u*diagm(s)*transpose(v)
end
#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% @testset "inv" begin
# inv
@net f(x) = inv(x) @net f(x) = inv(x)
m = tf(f) m = tf(f)
@test maximum(abs.(m(A)-inv(A))) < error_margin @test m(A) ≈ inv(A)
end
#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% @testset "det" begin
# det
@net f(x) = det(x) @net f(x) = det(x)
m = tf(f) m = tf(f)
@test maximum(abs.(m(A)-det(A))) < error_margin @test m(A) ≈ det(A)
end
#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
end
end end