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

Merge pull request #41 from alha02/add-more-tf-ops

Browse Source 
Add tf ops: svd, inv, det
This commit is contained in:
Mike J Innes 2017-06-09 11:20:42 +01:00 committed by GitHub
commit 0f433137d0
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/tensorflow/graph.jl
View File

@ -27,9 +27,11 @@ graph(::typeof(max), x, dim=nothing) = TensorFlow.reduce_max(x;axis=dim)
graph(::typeof(all), x, dim=nothing) = TensorFlow.reduce_all(x;axis=dim) graph(::typeof(all), x, dim=nothing) = TensorFlow.reduce_all(x;axis=dim)
graph(::typeof(any), x, dim=nothing) = TensorFlow.reduce_any(x;axis=dim) graph(::typeof(any), x, dim=nothing) = TensorFlow.reduce_any(x;axis=dim)
graph(::typeof(mean), x, dim=nothing) = TensorFlow.reduce_mean(x;axis=dim) graph(::typeof(mean), x, dim=nothing) = TensorFlow.reduce_mean(x;axis=dim)
graph(::typeof(svd), x) = svd(x)
for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos, for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos,
sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj) sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj,
inv, det)
@eval graph(::typeof($op), args...) = $op(args...) @eval graph(::typeof($op), args...) = $op(args...)
end end

8
test/backend/tensorflow.jl
View File

@ -23,4 +23,12 @@ test_anon(tf)
@test run(sess, Y, Dict(X=>xs)) ≈ d(xs) @test run(sess, Y, Dict(X=>xs)) ≈ d(xs)
end end
@testset "Ops" begin
A = randn(Float32,(5,5))
u,s,v = tf(@net x -> svd(x))(A)
@test A ≈ u*diagm(s)*transpose(v)
@test tf(@net x -> inv(x))(A) ≈ inv(A)
@test tf(@net x -> det(x))(A) ≈ det(A)
end
end end