TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix std arguments in normalise

Browse Source
This commit is contained in:
Moksh Jain 2019-02-05 18:36:04 +05:30
parent c6409d7686
commit 046f7b4eae
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/stateless.jl
View File

@ -46,6 +46,6 @@ logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ)
""" """
function normalise(x::AbstractArray, dims::Int=1) function normalise(x::AbstractArray, dims::Int=1)
μ′ = mean(x, dims = dims) μ′ = mean(x, dims = dims)
σ′ = std(x, dims = dims, mean = μ′) σ′ = std(x, dims = dims, mean = μ′, corrected=false)
return (x .- μ′) ./ σ′ return (x .- μ′) ./ σ′
end end