add reshape

src/Flux.jl

@@ -23,7 +23,7 @@ include("core.jl")
 import .FluxCore: back!, update!, graph
 
 include("utils.jl")
-
+include("ops.jl")
 include("params.jl")
 
 include("compiler/code.jl")

src/backend/tensorflow/graph.jl

@@ -31,6 +31,7 @@ graph(::typeof(svd), x) = svd(x)
 graph(::typeof(size), x, dim) = TensorFlow.size(x,convert(Tensor{Int32}, dim))
 graph(::typeof(size), x) = TensorFlow.size(x)
 graph(::typeof(chol), args...) = TensorFlow.transpose(TensorFlow.cholesky(args...))
+graph(::typeof(reshape), x, dims) = TensorFlow.reshape(x,convert(Tensor{Int32},dims))
 
 for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos,
            sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj,

src/ops.jl

@@ -0,0 +1,5 @@
+export reshape
+
+import Base: reshape
+
+reshape(x::AbstractArray, dims::AbstractArray) = reshape(x,tuple(dims...))

test/backend/tensorflow.jl

@@ -47,6 +47,8 @@ end
   A = randn(6,5)
   A = A'*A
   @test tf(@net x -> chol(x))(A) ≈ chol(A)
+  A = randn(Float32,(6,3))
+  @test transpose(tf(@net (x,y) -> reshape(x,y))(transpose(A),[2,9])) ≈ reshape(A,(9,2)) # Note: TF is row major and julia is not
 end
 
 end