diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl
index 4b868261..262a65c0 100644
--- a/src/layers/stateless.jl
+++ b/src/layers/stateless.jl
@@ -40,12 +40,17 @@ but it is more numerically stable.
 logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ)
 
 """
-    normalise(x::AbstractArray, dims::Int=1)
+    normalise(x::AbstractArray; dims::Int=1)
 
     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::Int=1)
   μ′ = mean(x, dims = dims)
   σ′ = std(x, dims = dims, mean = μ′, corrected=false)
   return (x .- μ′) ./ σ′
 end
+
+function normalise(x::AbstractArray, dims::Int=1)
+  Base.depwarn("`normalise(x::AbstractArray, dims)` is deprecated, use `normalise(a, dims=dims)` instead.", :normalise)
+  normalise(x, dims = dims)
+end