From e7783ace12908712342e5d0760cf3c6ff468888f Mon Sep 17 00:00:00 2001
From: James Bradbury <jekbradbury@gmail.com>
Date: Thu, 6 Sep 2018 18:38:11 -0700
Subject: [PATCH] 1.0 compat for `normalise`

---
 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 6d040f4f..891ec230 100644
--- a/src/layers/stateless.jl
+++ b/src/layers/stateless.jl
@@ -47,7 +47,7 @@ logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ)
 Normalise each column of `x` to mean 0 and standard deviation 1.
 """
 function normalise(x::AbstractVecOrMat)
-  μ′ = mean(x, 1)
-  σ′ = std(x, 1, mean = μ′)
+  μ′ = mean(x, dims = 1)
+  σ′ = std(x, dims = 1, mean = μ′)
   return (x .- μ′) ./ σ′
 end