Merge pull request #2072 from mcognetta/export_embedding

Export `Embedding`
FluxML · Oct 6, 2022 · 7a276f8 · 7a276f8
2 parents 66dddb9 + 3e8af9f
commit 7a276f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Flux.jl b/src/Flux.jl
@@ -16,7 +16,7 @@ export gradient
 # Pirate error to catch a common mistake. (Internal function `base` because overloading `update!` is more likely to give ambiguities.)
 Optimisers.base(dx::Zygote.Grads) = error("Optimisers.jl cannot be used with Zygote.jl's implicit gradients, `Params` & `Grads`")
 
-export Chain, Dense, Maxout, SkipConnection, Parallel, PairwiseFusion,
+export Chain, Dense, Embedding, Maxout, SkipConnection, Parallel, PairwiseFusion,
        RNN, LSTM, GRU, GRUv3,
        SamePad, Conv, CrossCor, ConvTranspose, DepthwiseConv,
        AdaptiveMaxPool, AdaptiveMeanPool, GlobalMaxPool, GlobalMeanPool, MaxPool, MeanPool,