This commit is contained in:
Johnny Chen 2018-08-23 21:56:32 +08:00
parent 6743d52d08
commit c9d6b5648f

View File

@ -79,7 +79,6 @@ function (a::Dense)(x::AbstractArray)
W, b, σ = a.W, a.b, a.σ
σ.(W*x .+ b)
end
(a::Dense)(x::Number) = a([x]) # prevent broadcasting of scalar
function Base.show(io::IO, l::Dense)
print(io, "Dense(", size(l.W, 2), ", ", size(l.W, 1))