From c3e04392d83fa457de891e63a45673bf32a5e38c Mon Sep 17 00:00:00 2001 From: pshashk Date: Fri, 8 Feb 2019 16:15:37 +0300 Subject: [PATCH] drop dims type restriction --- src/layers/stateless.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl index 262a65c0..66309327 100644 --- a/src/layers/stateless.jl +++ b/src/layers/stateless.jl @@ -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. """ -function normalise(x::AbstractArray; dims::Int=1) +function normalise(x::AbstractArray; dims=1) μ′ = mean(x, dims = dims) σ′ = std(x, dims = dims, mean = μ′, corrected=false) return (x .- μ′) ./ σ′ 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) normalise(x, dims = dims) end