From 697a300ec31225fcc9cd92148ebf96c588d154f2 Mon Sep 17 00:00:00 2001 From: Ali Hamdi Date: Fri, 9 Jun 2017 21:32:50 +0200 Subject: [PATCH] add diagm --- src/backend/tensorflow/graph.jl | 2 +- test/backend/tensorflow.jl | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/tensorflow/graph.jl b/src/backend/tensorflow/graph.jl index 32f1cf60..6c4ca052 100644 --- a/src/backend/tensorflow/graph.jl +++ b/src/backend/tensorflow/graph.jl @@ -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, transpose, permutedims, cat, length, diag) + inv, det, transpose, permutedims, cat, length, diag, diagm) @eval graph(::typeof($op), args...) = $op(args...) end diff --git a/test/backend/tensorflow.jl b/test/backend/tensorflow.jl index 1a41e5d8..b0de468b 100644 --- a/test/backend/tensorflow.jl +++ b/test/backend/tensorflow.jl @@ -39,6 +39,8 @@ end @test tf(@net x -> length(x))(A1) == length(A1) A = randn(Float32,(5,5)) @test tf(@net x -> diag(x))(A) ≈ diag(A) + A = randn(Float32,(5,)) + @test tf(@net x -> diagm(x))(A) ≈ diagm(A) end end