fix merge

FluxML · Jan 8, 2021 · 847efd2 · 847efd2
1 parent 65c37c1
commit 847efd2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/layers/basic.jl b/src/layers/basic.jl
@@ -119,11 +119,10 @@ end
 
 @functor Dense
 
-function (a::Dense)(x::AbstractVecOrMat)
+function (a::Dense)(x::AbstractArray)
   W, b, σ = a.W, a.b, a.σ
-  # reshape to handle dims > 1 as batch dimensions
   sz = size(x)
-  x = reshape(x, sz[1], :) 
+  x = reshape(x, sz[1], :) # reshape to handle dims > 1 as batch dimensions 
   x = σ.(W*x .+ b)
   return reshape(x, :, sz[2:end]...)
 end