update tests
This commit is contained in:
parent
6a3f02b8e8
commit
f399179c02
@ -24,27 +24,11 @@ test_anon(tf)
|
|||||||
end
|
end
|
||||||
|
|
||||||
@testset "Ops" begin
|
@testset "Ops" begin
|
||||||
|
A = convert(Array{Float32},randn(5,5))
|
||||||
@testset "svd" begin
|
u,s,v = tf(@net x -> svd(x))(A)
|
||||||
A = convert(Array{Float32},randn(5,5))
|
@test A ≈ u*diagm(s)*transpose(v)
|
||||||
@net f(x) = svd(x)
|
@test tf(@net x -> inv(x))(A) ≈ inv(A)
|
||||||
m = tf(f)
|
@test tf(@net x -> det(x))(A) ≈ det(A)
|
||||||
u,s,v = m(A)
|
|
||||||
@test A ≈ u*diagm(s)*transpose(v)
|
|
||||||
end
|
|
||||||
|
|
||||||
@testset "inv" begin
|
|
||||||
@net f(x) = inv(x)
|
|
||||||
m = tf(f)
|
|
||||||
@test m(A) ≈ inv(A)
|
|
||||||
end
|
|
||||||
|
|
||||||
@testset "det" begin
|
|
||||||
@net f(x) = det(x)
|
|
||||||
m = tf(f)
|
|
||||||
@test m(A) ≈ det(A)
|
|
||||||
end
|
|
||||||
|
|
||||||
end
|
end
|
||||||
|
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue
Block a user