Update src/layers/basic.jl

Co-authored-by: Michael Abbott <[email protected]>
FluxML · Mar 1, 2022 · f617aa6 · f617aa6
1 parent de43bf5
commit f617aa6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/layers/basic.jl b/src/layers/basic.jl
@@ -476,7 +476,8 @@ function Base.show(io::IO, m::Parallel)
 end
 
 """
-    Embedding(in => out; init=randn)
+    Embedding(in => out; init=randn32)
+    Embedding(weight::AbstractMatrix)
 
 A lookup table that stores embeddings of dimension `out` 
 for a vocabulary of size `in`.