Update src/layers/basic.jl

Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com>

Author: manikyabard

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`.