From 37798ca2273ec23f74b824d3c1549bf83cc8930d Mon Sep 17 00:00:00 2001
From: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
Date: Wed, 6 Mar 2024 08:02:53 -0600
Subject: [PATCH] public at-layer

---
 src/Flux.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Flux.jl b/src/Flux.jl
index 7d2fe7e98d..a8720b7905 100644
--- a/src/Flux.jl
+++ b/src/Flux.jl
@@ -38,7 +38,7 @@ export Chain, Dense, Embedding, Maxout, SkipConnection, Parallel, PairwiseFusion
   # layers
   Bilinear, Scale, dropout,
   # utils
-  outputsize, state, create_bias,
+  outputsize, state, create_bias, @layer,
 ))
 
 include("optimise/Optimise.jl")