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

drop dims type restriction

Browse Source
This commit is contained in:
pshashk 2019-02-08 16:15:37 +03:00 committed by GitHub
parent ae10421bfe
commit c3e04392d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -44,13 +44,13 @@ logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ)
Normalises x to mean 0 and standard deviation 1, across the dimensions given by dims. Defaults to normalising over columns. Normalises x to mean 0 and standard deviation 1, across the dimensions given by dims. Defaults to normalising over columns.
""" """
function normalise(x::AbstractArray; dims::Int=1) function normalise(x::AbstractArray; dims=1)
μ′ = mean(x, dims = dims) μ′ = mean(x, dims = dims)
σ′ = std(x, dims = dims, mean = μ′, corrected=false) σ′ = std(x, dims = dims, mean = μ′, corrected=false)
return (x .- μ′) ./ σ′ return (x .- μ′) ./ σ′
end end
function normalise(x::AbstractArray, dims::Int=1) function normalise(x::AbstractArray, dims=1)
Base.depwarn("`normalise(x::AbstractArray, dims)` is deprecated, use `normalise(a, dims=dims)` instead.", :normalise) Base.depwarn("`normalise(x::AbstractArray, dims)` is deprecated, use `normalise(a, dims=dims)` instead.", :normalise)
normalise(x, dims = dims) normalise(x, dims = dims)
end end