adapt function

FluxML · Oct 4, 2017 · 285179e · 285179e
1 parent d93dc96
commit 285179e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/NNlib.jl b/src/NNlib.jl
@@ -3,5 +3,6 @@ module NNlib
 export σ, relu, softmax
 
 include("activation.jl")
+include("adapt.jl")
 
 end # module
diff --git a/src/adapt.jl b/src/adapt.jl
@@ -0,0 +1,7 @@
+# This doesn't really belong here, but it's convenient.
+
+adapt_(T, x) = x
+
+adapt(T, x) = adapt_(T, x)
+
+adapt(T, x::RowVector) = RowVector(adapt(T, x.vec))