Skip to content

Commit

Permalink
Merge pull request #539 from Sleort/patch-1
Browse files Browse the repository at this point in the history
Fix `Dense` documentation
  • Loading branch information
avik-pal authored Mar 12, 2024
2 parents 8dd9bc2 + c2008e2 commit c27b9f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/layers/basic.jl
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,7 @@ end

"""
Dense(in_dims => out_dims, activation=identity; init_weight=glorot_uniform,
init_bias=zeros32, bias::Bool=true)
init_bias=zeros32, use_bias::Bool=true, allow_fast_activation::Bool=true)
Create a traditional fully connected layer, whose forward pass is given by:
`y = activation.(weight * x .+ bias)`
Expand Down

0 comments on commit c27b9f5

Please sign in to comment.