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

1.0 compat for normalise

Browse Source
This commit is contained in:
James Bradbury 2018-09-06 18:38:11 -07:00
parent 5e4ee827e9
commit e7783ace12
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/layers/stateless.jl
View File

@ -47,7 +47,7 @@ logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ)
Normalise each column of `x` to mean 0 and standard deviation 1. Normalise each column of `x` to mean 0 and standard deviation 1.
""" """
function normalise(x::AbstractVecOrMat) function normalise(x::AbstractVecOrMat)
μ′ = mean(x, 1) μ′ = mean(x, dims = 1)
σ′ = std(x, 1, mean = μ′) σ′ = std(x, dims = 1, mean = μ′)
return (x .- μ′) ./ σ′ return (x .- μ′) ./ σ′
end end