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

add permutedims

Browse Source
This commit is contained in:
Ali Hamdi 2017-06-09 08:25:47 +02:00
parent d78963ab27
commit 57c733af67
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,7 +31,7 @@ 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, transpose) inv, det, transpose, permutedims)
@eval graph(::typeof($op), args...) = $op(args...) @eval graph(::typeof($op), args...) = $op(args...)
end end

2
test/backend/tensorflow.jl
View File

@ -31,6 +31,8 @@ end
@test tf(@net x -> det(x))(A) ≈ det(A) @test tf(@net x -> det(x))(A) ≈ det(A)
A = randn(Float32,(6,3)) A = randn(Float32,(6,3))
@test tf(@net x -> transpose(x))(A) ≈ transpose(A) @test tf(@net x -> transpose(x))(A) ≈ transpose(A)
A = randn(Float32,(6,3,2))
@test tf(@net (x,y) -> permutedims(x,y))(A,[3,2,1]) ≈ permutedims(A,[3,2,1])
end end
end end