fix import

Author: CarloLucibello

src/layers/basic.jl

@@ -443,6 +443,8 @@ or the corresponding [onehot encoding](@ref Flux.OneHotArray).
 # Examples
 
 ```julia-repl
+julia> using Flux: Embedding
+
 julia> vocab_size, embed_size = 1000, 4;
 
 julia> model = Embedding(vocab_size, embed_size)

test/cuda/layers.jl

@@ -262,7 +262,7 @@ end
 
 @testset "Embedding" begin
   vocab_size, embed_size = 5, 2
-  m = Embedding(vocab_size, embed_size)
+  m = Flux.Embedding(vocab_size, embed_size)
 
   x = [1, 3, 5]
   y = m(x)
@@ -281,7 +281,7 @@ end
 
   @testset "repeated indexes" begin
     vocab_size, embed_size = 5, 2
-    m = Embedding(vocab_size, embed_size)
+    m = Flux.Embedding(vocab_size, embed_size)
 
     x = [1, 3, 5, 3] # repeated indexes
     y = m(x)

test/layers/basic.jl

@@ -194,7 +194,7 @@ import Flux: activations
 
   @testset "Embedding" begin
     vocab_size, embed_size = 10, 4
-    m = Embedding(vocab_size, embed_size)
+    m = Flux.Embedding(vocab_size, embed_size)
     @test size(m.weight) == (embed_size, vocab_size)
 
     x = rand(1:vocab_size, 3)