add transpose

This commit is contained in:
Ali Hamdi 2017-06-09 07:36:38 +02:00
parent 60e3a94952
commit d78963ab27
2 changed files with 3 additions and 1 deletions

View File

@ -31,7 +31,7 @@ graph(::typeof(svd), x) = svd(x)
for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos,
sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj,
inv, det)
inv, det, transpose)
@eval graph(::typeof($op), args...) = $op(args...)
end

View File

@ -29,6 +29,8 @@ end
@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)
A = randn(Float32,(6,3))
@test tf(@net x -> transpose(x))(A) transpose(A)
end
end