diff --git a/src/layers/basic.jl b/src/layers/basic.jl index 26b2daaf..f7c5a1bc 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -43,7 +43,7 @@ Creates a traditional `Dense` layer with parameters `W` and `b`. y = σ.(W * x .+ b) The input `x` must be a vector of length `in`, or a batch of vectors represented -as an `in × N` matrix. The out `y` will be a vector or batch of length `in`. +as an `in × N` matrix. The out `y` will be a vector or batch of length `out`. julia> d = Dense(5, 2) Dense(5, 2)