File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -172,7 +172,7 @@ function Base.show(io::IO, l::Dense)
172172end
173173
174174"""
175- Diagonal(size::Integer...; bias = true, init = ones32)
175+ Diagonal(size::Integer...; bias= true, init= ones32)
176176 Diagonal(scale::AbstractArray, [bias])
177177
178178Create an element-wise linear layer, which performs
@@ -181,9 +181,9 @@ Create an element-wise linear layer, which performs
181181
182182with no activation function.
183183
184- The learnable parameters are initialised `scale = init(size...)` and
185- `bias = zeros32(size...)`, with `init = ones32` by default. You may specify the function `init`,
186- turn off trainable bias with `bias = false`, or provide the array(s) explicitly.
184+ The learnable scale & bias are initialised `init(size...)` and `zeros32(size...)`,
185+ with `init= ones32` by default. You may specify the function `init`,
186+ turn off trainable bias with `bias= false`, or provide the array(s) explicitly.
187187
188188Used by [`LayerNorm`](@ref).
189189"""
You can’t perform that action at this time.
0 commit comments